Skip to content

Commit ac87b49

Browse files
enable model export (#3952)
Signed-off-by: Ashwin Vaidya <[email protected]>
1 parent aaa2765 commit ac87b49

File tree

2 files changed

+0
-14
lines changed

2 files changed

+0
-14
lines changed

src/otx/core/model/anomaly.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44

55
from __future__ import annotations
66

7-
import logging as log
87
from typing import TYPE_CHECKING, Any, TypeAlias
98

109
import torch
@@ -285,12 +284,6 @@ def export(
285284
Returns:
286285
Path: path to the exported model.
287286
"""
288-
if export_format == OTXExportFormatType.OPENVINO:
289-
if to_exportable_code:
290-
msg = "Exportable code option is not supported yet for anomaly tasks and will be ignored."
291-
log.warning(msg)
292-
to_exportable_code = False
293-
294287
return self._exporter.export(
295288
model=self.model,
296289
output_dir=output_dir,

tests/integration/cli/test_cli.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -158,15 +158,8 @@ def test_otx_e2e(
158158
ExportCase2Test("ONNX", False, "exported_model_decoder.onnx"),
159159
ExportCase2Test("OPENVINO", False, "exported_model_decoder.xml"),
160160
] # TODO (sungchul): EXPORTABLE_CODE will be supported
161-
elif "anomaly" in task:
162-
fxt_export_list = [
163-
ExportCase2Test("ONNX", False, "exported_model.onnx"),
164-
ExportCase2Test("OPENVINO", False, "exported_model.xml"),
165-
] # anomaly doesn't support exportable code
166161

167162
overrides = fxt_cli_override_command_per_task[task]
168-
if "anomaly" in task:
169-
overrides = {} # Overrides are not needed in export
170163

171164
tmp_path_test = tmp_path / f"otx_test_{model_name}"
172165
for export_case in fxt_export_list:

0 commit comments

Comments
 (0)