Fixed bug for loss in state file.

climax/core/dataset.py

@@ -124,13 +124,12 @@ class EoDataset(torch.utils.data.Dataset):
         state_file = '_'.join([state_file, 'doy']) if doy else state_file
 
         # check which loss function is used
-        if predictand == 'pr':
-            if (isinstance(loss, BernoulliGammaLoss) or
-                isinstance(loss, BernoulliWeibullLoss)):
-                # adjust state file for precipitation
-                state_file = '_'.join([state_file, '{}mm'.format(
-                    str(loss.min_amount).replace('.', '')),
-                    repr(loss).strip('()')])
+        if (isinstance(loss, BernoulliGammaLoss) or
+            isinstance(loss, BernoulliWeibullLoss)):
+            # adjust state file for precipitation
+            state_file = '_'.join([state_file, '{}mm'.format(
+                str(loss.min_amount).replace('.', '')),
+                repr(loss).strip('()')])
         else:
             # add name of loss function to state file
             state_file = '_'.join([state_file, repr(loss).strip('()')])