chore: inputs -> arg_inputs

pytorch · Mar 2, 2025 · 16d3ac8 · 16d3ac8
1 parent 21ab1b8
commit 16d3ac8
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/tests/py/dynamo/runtime/test_004_weight_streaming.py b/tests/py/dynamo/runtime/test_004_weight_streaming.py
@@ -291,10 +291,6 @@ def test_weight_streaming_cudagraphs(self, _, use_python_runtime):
             ("cpp_runtime", False),
         ]
     )
-    @unittest.skipIf(
-        os.environ.get("CI_BUILD") == "1",
-        "Skipping test due to CI resource constraints",
-    )
     def test_runtime_state_change(self, _, use_python_runtime):
         class SampleModel(torch.nn.Module):
             def __init__(self):
@@ -333,7 +329,7 @@ def forward(self, x, b=None, c=None, d=None, e=[]):
         kwarg_torchtrt_input = prepare_inputs(input_list[0][1])
 
         compile_spec = {
-            "inputs": [
+            "arg_inputs": [
                 torchtrt.Input(
                     min_shape=(1, 100),
                     opt_shape=(64, 100),