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):