@@ -33,8 +33,8 @@ def __init__(
3333 checkpoint_name : str = "model" ,
3434 save_weights_only : bool = False ,
3535 save_best_only : bool = False ,
36- inference_variables : Sequence [str ] | str = "theta" ,
37- inference_conditions : Sequence [str ] | str = "x" ,
36+ inference_variables : Sequence [str ] | str = None ,
37+ inference_conditions : Sequence [str ] | str = None ,
3838 summary_variables : Sequence [str ] | str = None ,
3939 standardize : Sequence [str ] | str = "inference_variables" ,
4040 ** kwargs ,
@@ -72,10 +72,10 @@ def __init__(
7272 as some losses (e.g. flow matching) do not reliably reflect model performance, and outliers in the
7373 validation data can cause unwanted effects.
7474 inference_variables : Sequence[str] or str, optional
75- Variables for inference as a sequence of strings or a single string (default is "theta" ).
75+ Variables for inference as a sequence of strings or a single string (default is None ).
7676 Important for automating diagnostics!
7777 inference_conditions : Sequence[str] or str, optional
78- Variables used as conditions for inference (default is "x" ).
78+ Variables used as conditions for inference (default is None ).
7979 summary_variables : Sequence[str] or str, optional
8080 Variables for summarizing data, if any (default is None).
8181 standardize : Sequence[str] or str, optional
0 commit comments