From 06a85aa4a150ba5e810394c222d348009bd7440c Mon Sep 17 00:00:00 2001 From: BujSet Date: Thu, 7 Aug 2025 23:02:48 +0000 Subject: [PATCH 1/2] typo fix --- examples/models/mobilenet_v2/model.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/examples/models/mobilenet_v2/model.py b/examples/models/mobilenet_v2/model.py index 5c2c7ff7016..13b79de3404 100644 --- a/examples/models/mobilenet_v2/model.py +++ b/examples/models/mobilenet_v2/model.py @@ -57,8 +57,7 @@ def get_example_inputs(self): input_tensor = preprocess(input_image) input_batch = input_tensor.unsqueeze(0) input_batch = (input_batch,) - return (torch.randn(tensor_size),) - + return input_batch class MV2UntrainedModel(EagerModelBase): def __init__(self): From 4b423b7b7c49391fc22276e5f54c831bb997e3f7 Mon Sep 17 00:00:00 2001 From: BujSet Date: Thu, 7 Aug 2025 23:24:20 +0000 Subject: [PATCH 2/2] Linting --- examples/models/mobilenet_v2/model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/models/mobilenet_v2/model.py b/examples/models/mobilenet_v2/model.py index 13b79de3404..32e82197e46 100644 --- a/examples/models/mobilenet_v2/model.py +++ b/examples/models/mobilenet_v2/model.py @@ -59,6 +59,7 @@ def get_example_inputs(self): input_batch = (input_batch,) return input_batch + class MV2UntrainedModel(EagerModelBase): def __init__(self): pass