CropHarvestMultiClassValidation expects sample_size to be an attribut…

…e and for it to be set to None
nasaharvest · Mar 28, 2024 · b66f097 · b66f097
1 parent 572be39
commit b66f097
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/presto/eval/cropharvest_eval.py b/presto/eval/cropharvest_eval.py
@@ -302,6 +302,7 @@ def __init__(
         dw_suffix = "_no_dynamic_world" if ignore_dynamic_world else ""
         self.name = f"CropHarvest_multiclass_global{name_suffix}{dw_suffix}_{seed}"
         self.seed = seed
+        self.sample_size = None
 
     @torch.no_grad()
     def evaluate(