Fix Issue #13 (#14)

* Fix #13. * Fix run-test for C++ coverage.
tohtsky · Jun 30, 2022 · 6b2c6d0 · 6b2c6d0
1 parent cbe8b70
commit 6b2c6d0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/.github/workflows/run-test.yml b/.github/workflows/run-test.yml
@@ -18,7 +18,8 @@ jobs:
           pip install --upgrade pip
           pip install numpy scipy pandas
           sudo apt-get install lcov
-          CXXFLAGS="-O0 -g -coverage" pip install -e .
+          FLAGS="-fprofile-arcs -ftest-coverage"
+          CFLAGS="$FLAGS" CXXFLAGS="$FLAGS" pip install -e .
       - name: Run pytest
         run: |
           pip install pytest pytest-cov pytest-mock

diff --git a/src/myfm/utils/encoders/categorical.py b/src/myfm/utils/encoders/categorical.py
@@ -57,6 +57,12 @@ def _get_index(self, x: T) -> Optional[int]:
                 return None
             raise
 
+    def __getitem__(self, x: T) -> int:
+        result = self._get_index(x)
+        if result is None:
+            raise KeyError(f"{x} not found.")
+        return result
+
     def names(self) -> List[str]:
         return [str(y) for y in self.values]