#145 related patch

nicolay-r · Jul 7, 2024 · 46e4a83 · 46e4a83
1 parent 67479da
commit 46e4a83
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/arelight/run/infer.py b/arelight/run/infer.py
@@ -319,7 +319,7 @@ def __entity_display_value(entity_value):
     settings.append({
         "labels_scaler": labels_scaler,
         # We provide this settings for inference.
-        "predict_filepath": collection_target_func(DataType.Test),
+        "predict_filepath": collection_target_func(DataType.Test) + predict_extension[args.inference_writer],
         "samples_io": sampling_engines_setup["arekit"]["samples_io"],
         "predict_reader": predict_readers[args.inference_writer]
     })