Update and rename config_creation.py to test_config_creation.py

giotto-ai · Nov 8, 2023 · 8610926 · 8610926
1 parent ffdbf73
commit 8610926
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/tests/config_creation.py b/tests/config_creation.py
diff --git a/tests/test_config_creation.py b/tests/test_config_creation.py
@@ -0,0 +1,10 @@
+from pipeline_tool.pipeline_config import PipelineConfig
+
+def test_config():
+    config = PipelineConfig(input_shape=[1,2], output_shape=[12], data_type="long")
+
+    print("A simple config as been created")
+
+    config.create_mha_conf_equal(nb_mha=2, num_heads=12, embed_dim=3, dropout=0.0, batch_first=True)
+
+    print("An MHA config as been added")