Fix 'NoneType' object has no attribute 'shape'

apache · Aug 3, 2022 · fd9d187 · fd9d187
1 parent 225f5e6
commit fd9d187
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/frontend/pytorch/test_forward.py b/tests/python/frontend/pytorch/test_forward.py
@@ -137,7 +137,7 @@ def verify_model(
     elif isinstance(input_data, list):
         baseline_model = model_name
         baseline_input = input_data
-    elif isinstance(input_data, torch.Tensor) or not input_data.shape:
+    elif isinstance(input_data, torch.Tensor) or not input_data:
         baseline_model = model_name
         baseline_input = [input_data]
     else: