fix sorting bug (#1601)

neptune-ai · Dec 18, 2023 · 9fba58f · 9fba58f
1 parent b0cf789
commit 9fba58f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/neptune/internal/backends/hosted_neptune_backend.py b/src/neptune/internal/backends/hosted_neptune_backend.py
@@ -1032,7 +1032,7 @@ def _get_file_set_download_request(self, container_id: str, container_type: Cont
     def _get_column_types(self, project_id: UniqueId, column: str, types: Optional[Iterable[str]] = None) -> List[Any]:
         params = {
             "projectIdentifier": project_id,
-            "search": f"/^{column}$/",  # exact regex match
+            "search": column,
             "type": types,
             "params": {},
             **DEFAULT_REQUEST_KWARGS,

diff --git a/tests/unit/neptune/new/internal/backends/test_hosted_client.py b/tests/unit/neptune/new/internal/backends/test_hosted_client.py
@@ -551,6 +551,10 @@ class DTO:
         {"entries": [DTO(type="float"), DTO(type="int")], "result": AttributeType.FLOAT.value},
         {"entries": [DTO(type="float"), DTO(type="int"), DTO(type="datetime")], "result": AttributeType.STRING.value},
         {"entries": [DTO(type="float"), DTO(type="int"), DTO(type="string")], "result": AttributeType.STRING.value},
+        {
+            "entries": [DTO(type="float"), DTO(type="int"), DTO(type="string", name="test_column_different")],
+            "result": AttributeType.FLOAT.value,
+        },
     ]
 
     # then