temporary fixes for 505 and 506

amaiya · Jul 22, 2023 · b291ef4 · b291ef4
1 parent cff0ea5
commit b291ef4
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -18,7 +18,8 @@ Most recent releases are shown at the top. Each release shows:
 - N/A
 
 ### fixed:
-- N/A
+- Temporarily pin to `scikit-learn<1.3` to avoid `eli5` import error (#505) 
+- Temporarily changed `generative_qa` imports to avoid `OPENAI_API_KEY error (#506)
 
 
 ## 0.37.3 (2023-07-22)

diff --git a/README.md b/README.md
@@ -17,7 +17,7 @@
 ```python
 import os
 os.environ['OPENAI_API_KEY'] = 'ENTER YOUR OPENAI API KEY HERE'
-from ktrain.text.qa import GenerativeQA
+from ktrain.text.qa.generative_qa import GenerativeQA
 genqa = GenerativeQA()
 genqa.add_doc(text=a_string_containing_text_of_your_document)
 print(genqa.query('What is ktrain?'))

diff --git a/examples/text/question_answering_with_openai.ipynb b/examples/text/question_answering_with_openai.ipynb
@@ -40,7 +40,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from ktrain .text.qa import GenerativeQA\n",
+    "from ktrain .text.qa.generative_qa import GenerativeQA\n",
     "genqa = GenerativeQA()"
    ]
   },

diff --git a/ktrain/text/qa/__init__.py b/ktrain/text/qa/__init__.py
@@ -1,2 +1 @@
 from .extractive_qa import ExtractiveQABase, AnswerExtractor, SimpleQA
-from .generative_qa import GenerativeQA
diff --git a/setup.py b/setup.py
@@ -45,7 +45,7 @@
     url="https://github.com/amaiya/ktrain",
     keywords=["tensorflow", "keras", "deep learning", "machine learning"],
     install_requires=[
-        "scikit-learn",
+        "scikit-learn<1.3",
         "matplotlib >= 3.0.0",
         "pandas >= 1.0.1",
         "fastprogress >= 0.1.21",