Missed update, related to #105.

nicolay-r · Jun 29, 2021 · 15ceb25 · 15ceb25
1 parent 753c2c0
commit 15ceb25
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/common/experiment/output/opinions/converter.py b/common/experiment/output/opinions/converter.py
@@ -2,6 +2,7 @@
 from arekit.common.experiment.input.readers.opinion import InputOpinionReader
 from arekit.common.experiment.output.base import BaseOutput
 from arekit.common.labels.scaler import BaseLabelScaler
+from arekit.common.model.labeling.modes import LabelCalculationMode
 from arekit.common.model.labeling.single import SingleLabelsHelper
 from arekit.common.opinions.base import Opinion
 
@@ -22,7 +23,7 @@ def iter_opinion_collections(output_filepath,
         assert(isinstance(output_filepath, unicode))
         assert(isinstance(opinions_reader, InputOpinionReader))
         assert(callable(create_opinion_collection_func))
-        assert(isinstance(label_calculation_mode, unicode))
+        assert(isinstance(label_calculation_mode, LabelCalculationMode))
         assert(isinstance(supported_labels, set) or supported_labels is None)
         assert(isinstance(output, BaseOutput))