From b4390dbe10b663489e71121b9232531a26a8ac36 Mon Sep 17 00:00:00 2001 From: Alessandro Pappalardo Date: Wed, 5 May 2021 19:25:03 +0100 Subject: [PATCH] Fix (export): wrap export method not export_onnx --- src/brevitas/export/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/brevitas/export/__init__.py b/src/brevitas/export/__init__.py index f8be17d31..876219779 100644 --- a/src/brevitas/export/__init__.py +++ b/src/brevitas/export/__init__.py @@ -10,19 +10,19 @@ def export_finn_onnx(*args, **kwargs): - return FINNManager.export_onnx(*args, **kwargs) + return FINNManager.export(*args, **kwargs) def export_dpuv1_onnx(*args, **kwargs): - return DPUv1Manager.export_onnx(*args, **kwargs) + return DPUv1Manager.export(*args, **kwargs) def export_dpuv2_onnx(*args, **kwargs): - return DPUv2Manager.export_onnx(*args, **kwargs) + return DPUv2Manager.export(*args, **kwargs) def export_standard_onnx(*args, **kwargs): - return StdONNXManager.export_onnx(*args, **kwargs) + return StdONNXManager.export(*args, **kwargs) def jit_trace_dpuv1(*args, **kwargs):