From 3679d536bd7a446af7205f967cc96b42eb1f90f2 Mon Sep 17 00:00:00 2001
From: Jingya HUANG <44135271+JingyaHuang@users.noreply.github.com>
Date: Tue, 22 Oct 2024 13:57:07 +0000
Subject: [PATCH] fix tests

---
 tests/exporters/test_export.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/exporters/test_export.py b/tests/exporters/test_export.py
index 84dfd8d3b..dcf2b09dd 100644
--- a/tests/exporters/test_export.py
+++ b/tests/exporters/test_export.py
@@ -152,7 +152,7 @@ def _neuronx_export(
         with NamedTemporaryFile("w") as output:
             try:
                 _, neuron_outputs = export(
-                    model=model,
+                    model_or_path=model,
                     config=neuron_config,
                     output=Path(output.name),
                     inline_weights_to_neff=inline_weights_to_neff,