diff --git a/pyproject.toml b/pyproject.toml index 76d9726..f05af7c 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -28,4 +28,4 @@ dependencies = [ [project.scripts] train_npe_model = "cryo_sbi.inference.command_line_tools:cl_npe_train_no_saving" -model_to_tensor = "cryo_sbi.utils.command_line_tools:cl_models_to_tensor" +models_to_tensor = "cryo_sbi.utils.command_line_tools:cl_models_to_tensor"