Skip to content

Commit 6367fb5

Browse files
beat-buesserabigailgold
authored andcommitted
Fix style checks
Signed-off-by: Beat Buesser <[email protected]>
1 parent 31e34fe commit 6367fb5

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

utils/resources/create_model_weights.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -140,18 +140,13 @@ def create_scikit_model_weights():
140140
"linearSVC": LinearSVC(),
141141
}
142142

143-
clipped_models = {
144-
model_name: model for model_name, model in model_list.items()
145-
}
143+
clipped_models = {model_name: model for model_name, model in model_list.items()}
146144
unclipped_models = {model_name: model for model_name, model in model_list.items()}
147145

148146
(x_train_iris, y_train_iris), (_, _), _, _ = load_dataset("iris")
149147

150148
y_train_iris = np.argmax(y_train_iris, axis=1)
151149

152-
print(sklearn.__version__)
153-
dfg
154-
155150
for model_name, model in clipped_models.items():
156151
model.fit(X=x_train_iris, y=y_train_iris)
157152
pickle.dump(

0 commit comments

Comments
 (0)