try different method to exclude experimental from cov test

mitre · Oct 19, 2023 · e619b65 · e619b65
1 parent 29303ed
commit e619b65
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -30,4 +30,4 @@ jobs:
     - name: Unit and coverage tests
       run: |
         pytest tests/menelaus --cov=menelaus/ --cov-report term
-        coverage report -m --fail-under=100
+        coverage report -m --fail-under=100 --omit=menelaus/experimental
diff --git a/menelaus/experimental/transform.py b/menelaus/experimental/transform.py
@@ -1,6 +1,6 @@
 import numpy as np
 import tensorflow as tf
-from tensorflow.keras.layers import Flatten, Dense
+from tensorflow.keras.layers import Flatten, Dense, InputLayer
 from toolz import curry
 from transformers import AutoConfig, AutoTokenizer, TFAutoModel
 from typing import Optional, Dict, Union, List