Skip to content

Commit 77c4241

Browse files
committed
chore: inputs -> arg_inputs
1 parent abc915a commit 77c4241

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

tests/py/dynamo/runtime/test_004_weight_streaming.py

+1-5
Original file line numberDiff line numberDiff line change
@@ -291,10 +291,6 @@ def test_weight_streaming_cudagraphs(self, _, use_python_runtime):
291291
("cpp_runtime", False),
292292
]
293293
)
294-
@unittest.skipIf(
295-
os.environ.get("CI_BUILD") == "1",
296-
"Skipping test due to CI resource constraints",
297-
)
298294
def test_runtime_state_change(self, _, use_python_runtime):
299295
class SampleModel(torch.nn.Module):
300296
def __init__(self):
@@ -333,7 +329,7 @@ def forward(self, x, b=None, c=None, d=None, e=[]):
333329
kwarg_torchtrt_input = prepare_inputs(input_list[0][1])
334330

335331
compile_spec = {
336-
"inputs": [
332+
"arg_inputs": [
337333
torchtrt.Input(
338334
min_shape=(1, 100),
339335
opt_shape=(64, 100),

0 commit comments

Comments
 (0)