commit

Author: michaelcao28

pymc_experimental/inference/pathfinder.py

@@ -99,16 +99,18 @@ def fit_pathfinder(
 
     model = modelcontext(model)
 
-    ip = model.initial_point()
+    ip = model.initial_point()  # DEV: testing
     ip_map = DictToArrayBijection.map(ip)
 
     new_logprob, new_input = pm.pytensorf.join_nonshared_inputs(
         ip, (model.logp(),), model.value_vars, ()
     )
 
     logprob_fn_list = get_jaxified_graph([new_input], new_logprob)
+    # DEV: testing
 
     def logprob_fn(x):
+        # DEV: testing
         return logprob_fn_list(x)[0]
 
     [pathfinder_seed, sample_seed] = _get_seeds_per_chain(random_seed, 2)