black .

mlcommons · Jan 18, 2024 · 0853b68 · 0853b68
1 parent 2b69fa8
commit 0853b68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GANDLF/inference_manager.py b/GANDLF/inference_manager.py
@@ -38,7 +38,7 @@ def InferenceManager(dataframe, modelDir, parameters, device, outputDir=None):
 
     # initialize parameters for inference
     for key in ["penalty_weights", "sampling_weights", "class_weights"]:
-        parameters[key] = parameters.get(key, None)    
+        parameters[key] = parameters.get(key, None)
 
     n_folds = parameters["nested_training"]["validation"]