fixing flake8

deepskies · Oct 23, 2024 · 0cd9988 · 0cd9988
1 parent 0cd9d27
commit 0cd9988
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/scripts/DeepEnsemble.py b/src/scripts/DeepEnsemble.py
@@ -518,7 +518,7 @@ def main():
     print("model name is ", model_name)
     print("dim is ", dim)
     print("norm params", norm_params)
-    model_ensemble = train.train_DE(
+    train.train_DE(
         trainDataLoader,
         x_val,
         y_val,

diff --git a/src/scripts/DeepEvidentialRegression.py b/src/scripts/DeepEvidentialRegression.py
@@ -460,7 +460,7 @@ def main():
         data_type=dim,
     )
     print("model name is ", model_name)
-    model = train.train_DER(
+    train.train_DER(
         trainDataLoader,
         x_val,
         y_val,
@@ -499,4 +499,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()