write_file_metadata_json - added optional parameter for pytorch models

sassoftware · Jan 23, 2024 · ae049df · ae049df
1 parent 1cbbdc0
commit ae049df
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/sasctl/pzmm/write_json_files.py b/src/sasctl/pzmm/write_json_files.py
@@ -510,6 +510,7 @@ def write_file_metadata_json(
             dict_list.append({"role": "scoreResource", "name": model_prefix + ".mojo"})
         elif is_tf_keras_model:
             dict_list.append({"role": "scoreResource", "name": model_prefix + ".h5"})
+        #conditions added for pytorch.
         elif is_pytorch_model:
             dict_list.append(({"role": "scoreResource", "name": model_prefix + ".pth"}))
         else: