r

AIStream-Peelout · Nov 1, 2023 · 5af2b62 · 5af2b62
1 parent 58b870a
commit 5af2b62
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/flood_forecast/pytorch_training.py b/flood_forecast/pytorch_training.py
@@ -16,6 +16,19 @@
 
 
 def multi_crit(crit_multi: List, output, labels, valid=None):
+    """_summary_
+
+    :param crit_multi: _description_
+    :type crit_multi: List
+    :param output: _description_
+    :type output: _type_
+    :param labels: _description_
+    :type labels: _type_
+    :param valid: _description_, defaults to None
+    :type valid: _type_, optional
+    :return: _description_
+    :rtype: _type_
+    """
     i = 0
     loss = 0.0
     for crit in crit_multi:

diff --git a/tests/test_series_id.py b/tests/test_series_id.py
@@ -1,5 +1,5 @@
 from flood_forecast.preprocessing.pytorch_loaders import CSVSeriesIDLoader, SeriesIDTestLoader
-from flood_forecast.evaluator import infer_on_torch_model
+# from flood_forecast.evaluator import infer_on_torch_model
 import unittest
 import os
 from torch.nn import MSELoss