InferenceGym: Reexport Dirichlet target.

Also, add the correct dependency for the numpy TFP.

PiperOrigin-RevId: 753666635

Author: SiegeLordEx

spinoffs/inference_gym/inference_gym/targets/BUILD

@@ -138,7 +138,7 @@ py_library(
     srcs = ["dirichlet.py"],
     deps = [
         ":model",
-        # tensorflow_probability dep,
+        "//tensorflow_probability:numpy",
     ],
 )
 

spinoffs/inference_gym/inference_gym/targets/__init__.py

@@ -23,6 +23,7 @@
   from inference_gym.targets.brownian_motion import BrownianMotionMissingMiddleObservations
   from inference_gym.targets.brownian_motion import BrownianMotionUnknownScales
   from inference_gym.targets.brownian_motion import BrownianMotionUnknownScalesMissingMiddleObservations
+  from inference_gym.targets.dirichlet import Dirichlet
   from inference_gym.targets.eight_schools import EightSchools
   from inference_gym.targets.ill_conditioned_gaussian import IllConditionedGaussian
   from inference_gym.targets.item_response_theory import ItemResponseTheory
@@ -69,6 +70,7 @@
     'BrownianMotionUnknownScalesMissingMiddleObservations',
     'ConvectionLorenzBridge',
     'ConvectionLorenzBridgeUnknownScales',
+    'Dirichlet',
     'EightSchools',
     'GermanCreditNumericLogisticRegression',
     'GermanCreditNumericProbitRegression',