write_file_metadata_json - added optional parameter for pytorch models

namera9 · namera9 · commit ae049dfc4c47 · 2024-01-23T09:19:03.000-05:00
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:
