Fix fetch_runs_table method for state param usage

CHANGELOG.md

@@ -19,8 +19,9 @@
 - Removed `get_run_url` method ([#1238](https://github.com/neptune-ai/neptune-client/pull/1238))
 - Removed `neptune.new.sync` module ([#1240](https://github.com/neptune-ai/neptune-client/pull/1240))
 - Added exception for unsupported types ([#1229](https://github.com/neptune-ai/neptune-client/pull/1229))
-- Change run status to Active / Inactive ([#1233](https://github.com/neptune-ai/neptune-client/pull/1233))
+- Change run status in the table returned by `fetch_runs_table` to Active / Inactive ([#1233](https://github.com/neptune-ai/neptune-client/pull/1233))
 - Package renamed from `neptune-client` to `neptune` ([#1225](https://github.com/neptune-ai/neptune-client/pull/1225))
+- Changed values used to filter runs table by state ([#1253](https://github.com/neptune-ai/neptune-client/pull/1253))
 
 ## neptune-client 0.16.18
 

src/neptune/internal/utils/run_state.py

@@ -27,6 +27,13 @@ class RunState(enum.Enum):
     _api_active = "running"
     _api_inactive = "idle"
 
+    @classmethod
+    def from_string(cls, value: str) -> "RunState":
+        try:
+            return cls(value.capitalize())
+        except ValueError as e:
+            raise NeptuneException(f"Can't map RunState to API: {value}") from e
+
     @staticmethod
     def from_api(value: str) -> "RunState":
         if value == RunState._api_active.value:
@@ -35,3 +42,9 @@ def from_api(value: str) -> "RunState":
             return RunState.inactive
         else:
             raise NeptuneException(f"Unknown RunState: {value}")
+
+    def to_api(self) -> str:
+        if self is RunState.active:
+            return self._api_active.value
+        if self is RunState.inactive:
+            return self._api_inactive.value

src/neptune/metadata_containers/project.py

@@ -42,6 +42,7 @@
 from neptune.internal.operation_processors.operation_processor import OperationProcessor
 from neptune.internal.state import ContainerState
 from neptune.internal.utils import as_list
+from neptune.internal.utils.run_state import RunState
 from neptune.metadata_containers import MetadataContainer
 from neptune.metadata_containers.metadata_containers_table import Table
 from neptune.types.mode import Mode
@@ -131,7 +132,7 @@ def _prepare_nql_query(ids, states, owners, tags):
                             name="sys/state",
                             type=NQLAttributeType.EXPERIMENT_STATE,
                             operator=NQLAttributeOperator.EQUALS,
-                            value=state,
+                            value=RunState.from_string(state).to_api(),
                         )
                         for state in states
                     ],
@@ -261,7 +262,6 @@ def fetch_runs_table(
         tags = as_list("tag", tag)
 
         nql_query = self._prepare_nql_query(ids, states, owners, tags)
-
         return MetadataContainer._fetch_entries(
             self,
             child_type=ContainerType.RUN,

tests/e2e/standard/test_fetch_tables.py

@@ -20,6 +20,7 @@
 import pytest
 
 import neptune
+from neptune.exceptions import NeptuneException
 from neptune.metadata_containers import Model
 from tests.e2e.base import (
     BaseE2ETest,
@@ -155,3 +156,25 @@ def get_model_versions_as_rows(**kwargs):
             return container.fetch_model_versions_table(**kwargs).to_rows()
 
         self._test_fetch_from_container(init_run, get_model_versions_as_rows)
+
+    @pytest.mark.parametrize(
+        "state_active,state_inactive", [("active", "inactive"), ("Active", "Inactive"), ("aCTive", "inAcTiVE")]
+    )
+    def test_fetch_runs_table_with_correct_states(self, state_active, state_inactive, environment, project):
+        tag = str(uuid.uuid4())
+        with neptune.init_run(project=environment.project) as run:
+            run["sys/tags"] = tag
+            runs = project.fetch_runs_table(state=state_active).to_pandas()
+            assert not runs.empty
+            assert tag in runs["sys/tags"].values
+
+        time.sleep(5)
+
+        runs = project.fetch_runs_table(state=state_inactive).to_pandas()
+        assert not runs.empty
+        assert tag in runs["sys/tags"].values
+
+    def test_fetch_runs_table_with_incorrect_states(self, project):
+        for incorrect_state in ["idle", "running", "some_arbitrary_state"]:
+            with pytest.raises(NeptuneException):
+                project.fetch_runs_table(state=incorrect_state)