specifies numpy

.github/workflows/test_gedi.yml

@@ -29,7 +29,7 @@ jobs:
         python -m pip install --upgrade pip
         pip install .
 
-    - name: Run feature_extraction test
+    - name: Run test
       run:
         python main.py -o config_files/options/baseline.json -a config_files/algorithm/feature_extraction.json
 
@@ -57,7 +57,7 @@ jobs:
         python -m pip install --upgrade pip
         pip install .
 
-    - name: Run feature_extraction test
+    - name: Run test
       run:
         python main.py -o config_files/options/baseline.json -a config_files/algorithm/generation.json
 
@@ -85,9 +85,34 @@ jobs:
         python -m pip install --upgrade pip
         pip install .
 
-    - name: Run feature_extraction test
+    - name: Run test
       run:
         python main.py -o config_files/options/baseline.json -a config_files/algorithm/benchmark.json
 
     - name: Compare output
       run: diff output/benchmark/test_2_benchmark.csv output/benchmark/test_2_benchmark.csv
+
+  test_augmentation:
+    runs-on: ubuntu-latest
+
+    # Setting up a python envronment  for the test script to run
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v2
+
+    - name: Set up Python
+      uses: actions/setup-python@v2
+      with:
+        python-version: 3.x
+
+    - name: Install dependencies
+      run: |
+        sudo apt install python3-packaging
+    - name: Install feeed
+      run: |
+        python -m pip install --upgrade pip
+        pip install .
+
+    - name: Run test
+      run:
+        python main.py -o config_files/options/baseline.json -a config_files/algorithm/augmentation.json

config_files/algorithm/generation.json

@@ -3,12 +3,7 @@
     "pipeline_step": "event_logs_generation",
     "output_path": "output",
     "generator_params": {
-      "experiment": {
-        "input_path": "data/grid_objectives.csv",
-        "objectives": ["epa_normalized_variant_entropy"],
-        "objectives": ["ratio_most_common_variant", "epa_normalized_sequence_entropy"], 
-        "objectives": ["ratio_top_20_variants","epa_normalized_sequence_entropy_linear_forgetting"]
-      },
+      "experiment": {"epa_normalized_sequence_entropy_linear_forgetting": 0.4, "ratio_top_20_variants": 0.2},
       "config_space": {
         "mode": [5, 20],
         "sequence": [0.01, 1],
@@ -21,7 +16,7 @@
         "duplicate": [0],
         "or": [0]
       },
-      "n_trials": 50,
+      "n_trials": 5,
       "plot_reference_feature": "epa_normalized_sequence_entropy",
       "plot_reference_feature": "epa_normalized_sequence_entropy_linear_forgetting",
       "plot_reference_feature": ""

gedi/generator.py

@@ -241,7 +241,6 @@ class GenerateEventLogs():
                     configspace_tuples[k] = v[0]
                 else:
                     configspace_tuples[k] = tuple(v)
-            print(configspace_tuples)
             configspace = ConfigurationSpace(configspace_tuples)
 
         if self.params.get(N_TRIALS) is None:

gedi/plotter.py

@@ -618,7 +618,7 @@ class AugmentationPlotter(object):
         if output_path != None:
             output_path += f"/augmentation_pca_{n_features}_{self.sampler}.jpg"
             fig.savefig(output_path)
-            print("SUCCESS: Saved augmentation pca plot at {output_path}")
+            print(f"SUCCESS: Saved augmentation pca plot at {output_path}")
 
 
 class GenerationPlotter(object):
@@ -673,7 +673,7 @@ class GenerationPlotter(object):
                 targets = orig_targets.copy()
             elif isinstance(orig_targets, defaultdict):
                 if k not in orig_targets:
-                    print("[WARNING] {k} not in targets. Only in generated features. Will continue with next feature to compare with")
+                    print(f"[WARNING] {k} not in targets. Only in generated features. Will continue with next feature to compare with")
                     continue
                 targets = orig_targets[k].copy()
             else:

setup.py

@@ -24,7 +24,7 @@ setup(
             'imblearn==0.0',
             'Levenshtein==0.23.0',
             'matplotlib==3.8.4',
-            'numpy>=1.23.1',
+            'numpy==1.26.4',
             'pandas==2.2.2',
             'pm4py==2.7.2',
             'scikit-learn==1.2.2',