Skip to content

Commit 3400af1

Browse files
committed
Black reformatting
1 parent d9f39c7 commit 3400af1

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

src/sasctl/pzmm/modelParameters.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import json
55

66
from .._services.model_repository import ModelRepository as mr
7+
78
# TODO: Convert STRINGIO calls to string or dict format
89

910

@@ -31,7 +32,9 @@ def _find_file(model, file_name):
3132
for file in file_list:
3233
print(file.name)
3334
if file_name.lower() in file.name.lower():
34-
correct_file = sess.get(f"modelRepository/models/{model}/contents/{file.id}/content")
35+
correct_file = sess.get(
36+
f"modelRepository/models/{model}/contents/{file.id}/content"
37+
)
3538
return correct_file
3639

3740

@@ -57,13 +60,16 @@ def generate_hyperparameters(cls, model, model_prefix, pickle_path):
5760
JSON file
5861
Named {model_prefix}Hyperparameters.json.
5962
"""
63+
6064
def sklearn_params():
6165
"""
6266
Generates hyperparameters for the models generated by scikit-learn.
6367
"""
6468
hyperparameters = model.get_params()
6569
model_json = {"hyperparameters": hyperparameters}
66-
with open(Path(pickle_path) / f"{model_prefix}Hyperparameters.json", "w") as f:
70+
with open(
71+
Path(pickle_path) / f"{model_prefix}Hyperparameters.json", "w"
72+
) as f:
6773
f.write(json.dumps(model_json, indent=4))
6874

6975
if all(hasattr(model, attr) for attr in ["_estimator_type", "get_params"]):
@@ -162,5 +168,5 @@ def add_hyperparameters(cls, model, **kwargs):
162168
mr.add_model_content(
163169
model,
164170
StringIO(json.dumps(hyperparameters, indent=4)),
165-
f"{model.name}Hyperparameters.json"
171+
f"{model.name}Hyperparameters.json",
166172
)

0 commit comments

Comments
 (0)