Annot might be None (futher generalization)

nicolay-r · Jun 15, 2022 · c046909 · c046909
1 parent d10ef20
commit c046909
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/arekit/common/experiment/api/ctx_serialization.py b/arekit/common/experiment/api/ctx_serialization.py
@@ -9,7 +9,7 @@ class ExperimentSerializationContext(ExperimentContext):
 
     def __init__(self, label_scaler, annot, name_provider, data_folding):
         assert(isinstance(label_scaler, BaseLabelScaler))
-        assert(isinstance(annot, BaseAnnotator))
+        assert(isinstance(annot, BaseAnnotator) or annot is None)
         super(ExperimentSerializationContext, self).__init__(name_provider=name_provider,
                                                              data_folding=data_folding)