default value for tests

openvinotoolkit · Jan 10, 2025 · 7ceecf3 · 7ceecf3
1 parent 05a8142
commit 7ceecf3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/anomalib/models/components/base/export_mixin.py b/src/anomalib/models/components/base/export_mixin.py
@@ -80,7 +80,7 @@ class ExportMixin:
     def to_torch(
         self,
         export_root: Path | str,
-        model_file_name: str,
+        model_file_name: str = "model",
     ) -> Path:
         """Export model to PyTorch format.
 
@@ -111,7 +111,7 @@ def to_torch(
     def to_onnx(
         self,
         export_root: Path | str,
-        model_file_name: str,
+        model_file_name: str = "model",
         input_size: tuple[int, int] | None = None,
     ) -> Path:
         """Export model to ONNX format.
@@ -166,7 +166,7 @@ def to_onnx(
     def to_openvino(
         self,
         export_root: Path | str,
-        model_file_name: str,
+        model_file_name: str = "model",
         input_size: tuple[int, int] | None = None,
         compression_type: CompressionType | None = None,
         datamodule: AnomalibDataModule | None = None,