fix StrEnum issue

Author: jjmachan

nbs/backends/ragas_api_client.ipynb

@@ -44,12 +44,8 @@
     "#| export\n",
     "import httpx\n",
     "import asyncio\n",
-    "import functools\n",
     "import typing as t\n",
-    "import inspect\n",
     "from pydantic import BaseModel, Field\n",
-    "from enum import StrEnum\n",
-    "import uuid\n",
     "from fastcore.utils import patch"
    ]
   },
@@ -708,7 +704,17 @@
    "outputs": [],
    "source": [
     "#| export\n",
-    "class ColumnType(StrEnum):\n",
+    "from enum import Enum"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#| export\n",
+    "class ColumnType(str, Enum):\n",
     "    NUMBER = \"number\"\n",
     "    TEXT = \"text\"\n",
     "    LONG_TEXT = \"longText\"\n",

ragas_annotator/backends/ragas_api_client.py

@@ -8,12 +8,8 @@
 # %% ../../nbs/backends/ragas_api_client.ipynb 4
 import httpx
 import asyncio
-import functools
 import typing as t
-import inspect
 from pydantic import BaseModel, Field
-from enum import StrEnum
-import uuid
 from fastcore.utils import patch
 
 # %% ../../nbs/backends/ragas_api_client.ipynb 5
@@ -262,7 +258,10 @@ async def delete_experiment(self: RagasApiClient, project_id: str, experiment_id
 
 
 # %% ../../nbs/backends/ragas_api_client.ipynb 25
-class ColumnType(StrEnum):
+from enum import Enum
+
+# %% ../../nbs/backends/ragas_api_client.ipynb 26
+class ColumnType(str, Enum):
     NUMBER = "number"
     TEXT = "text"
     LONG_TEXT = "longText"
@@ -272,7 +271,7 @@ class ColumnType(StrEnum):
     CHECKBOX = "checkbox"
     CUSTOM = "custom"
 
-# %% ../../nbs/backends/ragas_api_client.ipynb 26
+# %% ../../nbs/backends/ragas_api_client.ipynb 27
 #---- Dataset Columns ----
 @patch
 async def list_dataset_columns(
@@ -343,7 +342,7 @@ async def delete_dataset_column(
         f"projects/{project_id}/datasets/{dataset_id}/columns/{column_id}"
     )
 
-# %% ../../nbs/backends/ragas_api_client.ipynb 34
+# %% ../../nbs/backends/ragas_api_client.ipynb 35
 #---- Dataset Rows ----
 @patch
 async def list_dataset_rows(
@@ -405,11 +404,11 @@ async def delete_dataset_row(
     )
 
 
-# %% ../../nbs/backends/ragas_api_client.ipynb 46
+# %% ../../nbs/backends/ragas_api_client.ipynb 47
 import uuid
 import string
 
-# %% ../../nbs/backends/ragas_api_client.ipynb 47
+# %% ../../nbs/backends/ragas_api_client.ipynb 48
 def create_nano_id(size=12):
     # Define characters to use (alphanumeric)
     alphabet = string.ascii_letters + string.digits
@@ -426,11 +425,11 @@ def create_nano_id(size=12):
     # Pad if necessary and return desired length
     return result[:size]
 
-# %% ../../nbs/backends/ragas_api_client.ipynb 49
+# %% ../../nbs/backends/ragas_api_client.ipynb 50
 import uuid
 import string
 
-# %% ../../nbs/backends/ragas_api_client.ipynb 50
+# %% ../../nbs/backends/ragas_api_client.ipynb 51
 def create_nano_id(size=12):
     # Define characters to use (alphanumeric)
     alphabet = string.ascii_letters + string.digits
@@ -447,7 +446,7 @@ def create_nano_id(size=12):
     # Pad if necessary and return desired length
     return result[:size]
 
-# %% ../../nbs/backends/ragas_api_client.ipynb 52
+# %% ../../nbs/backends/ragas_api_client.ipynb 53
 # Default settings for columns
 DEFAULT_SETTINGS = {
     "is_required": False,
@@ -470,7 +469,7 @@ class Row(BaseModel):
     id: str = Field(default_factory=create_nano_id)
     data: t.List[RowCell] = Field(...)
 
-# %% ../../nbs/backends/ragas_api_client.ipynb 53
+# %% ../../nbs/backends/ragas_api_client.ipynb 54
 #---- Resource With Data Helper Methods ----
 @patch
 async def _create_with_data(
@@ -597,7 +596,7 @@ async def create_dataset_with_data(
         "dataset", project_id, name, description, columns, rows, batch_size
     )
 
-# %% ../../nbs/backends/ragas_api_client.ipynb 59
+# %% ../../nbs/backends/ragas_api_client.ipynb 60
 #---- Experiment Columns ----
 @patch
 async def list_experiment_columns(
@@ -728,7 +727,7 @@ async def delete_experiment_row(
         f"projects/{project_id}/experiments/{experiment_id}/rows/{row_id}"
     )
 
-# %% ../../nbs/backends/ragas_api_client.ipynb 62
+# %% ../../nbs/backends/ragas_api_client.ipynb 63
 @patch
 async def create_experiment_with_data(
     self: RagasApiClient,
@@ -759,7 +758,7 @@ async def create_experiment_with_data(
         "experiment", project_id, name, description, columns, rows, batch_size
     )
 
-# %% ../../nbs/backends/ragas_api_client.ipynb 63
+# %% ../../nbs/backends/ragas_api_client.ipynb 64
 #---- Utility Methods ----
 @patch
 def create_column(