Update and rename trace_complex.py to test_trace_complex.py

giotto-ai · Nov 8, 2023 · 4100e5c · 4100e5c
1 parent 0b9b2e0
commit 4100e5c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/trace_complex.py → tests/test_trace_complex.py b/tests/trace_complex.py → tests/test_trace_complex.py
@@ -31,4 +31,5 @@ def forward(self, image):
 
 config_pipeline.create_mha_conf_equal(nb_mha, num_heads, embed_dim, dropout, batch_first)
 
-trace = SkippableTracing(nb_gpus=0, model=model, config=config_pipeline)
+def test_trace():
+    trace = SkippableTracing(nb_gpus=0, model=model, config=config_pipeline)