fixed mlp_mixer variable name bug

tenstorrent · Jan 31, 2025 · 94ab912 · 94ab912
1 parent 9dee673
commit 94ab912
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/forge/test/mlir/operators/tm/test_in_place.py b/forge/test/mlir/operators/tm/test_in_place.py
@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: (c) 2025 Tenstorrent AI ULC
+#
+# SPDX-License-Identifier: Apache-2.0
 import pytest
 import torch
 import tensorflow as tf

diff --git a/forge/test/models/pytorch/vision/mlp_mixer/test_mlp_mixer.py b/forge/test/models/pytorch/vision/mlp_mixer/test_mlp_mixer.py
@@ -49,7 +49,7 @@ def test_mlp_mixer_timm_pytorch(record_forge_property, variant):
     record_forge_property("model_name", module_name)
 
     framework_model = download_model(timm.create_model, variant, pretrained=True)
-    config = resolve_data_config({}, model=model)
+    config = resolve_data_config({}, model=framework_model)
     transform = create_transform(**config)
 
     try: