Format

protectai · Nov 30, 2023 · f06cee5 · f06cee5
1 parent f96a1e0
commit f06cee5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modelscan/models/h5/scan.py b/modelscan/models/h5/scan.py
@@ -55,7 +55,7 @@ def _get_keras_h5_operator_names(source: Union[str, Path]) -> List[str]:
         # Todo: source isn't guaranteed to be a file
         with h5py.File(source, "r") as model_hdf5:
             try:
-                model_config = json.loads(model_hdf5.attrs.get("model_config",{}))
+                model_config = json.loads(model_hdf5.attrs.get("model_config", {}))
                 layers = model_config.get("config", {}).get("layers", {})
                 lambda_layers = []
                 for layer in layers: