Update and rename trace_simple.py to test_trace_simple.py

giotto-ai · Nov 8, 2023 · 16556b6 · 16556b6
1 parent 4100e5c
commit 16556b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/trace_simple.py → tests/test_trace_simple.py b/tests/trace_simple.py → tests/test_trace_simple.py
@@ -34,5 +34,5 @@ def forward(self, x):
 config_pipeline = PipelineConfig(input_shape=[batch_size, 3, 32, 32],
                                  output_shape=[batch_size],
                                  data_type="long")
-
-trace = SkippableTracing(nb_gpus=0, model=model, config=config_pipeline)
+def test_trace_simple():
+    trace = SkippableTracing(nb_gpus=0, model=model, config=config_pipeline)