Merge branch 'develop'

CHANGELOG.md

@@ -6,6 +6,20 @@ Most recent releases are shown at the top. Each release shows:
 - **Changed**: Additional parameters, changes to inputs or outputs, etc
 - **Fixed**: Bug fixes that don't change documented behaviour
 
+## 0.27.3 (2021-09-03)
+
+### New:
+- N/A
+
+### Changed
+-N/A
+
+### Fixed:
+- `SimpleQA` now can load PyTorch question-answering checkpoints
+- change API call to support newest `causalnlp`
+
+
+
 ## 0.27.2 (2021-07-28)
 
 ### New:

ktrain/tabular/causalinference.py

@@ -57,7 +57,7 @@ def causal_inference_model(
         import causalnlp
     except ImportError:
         raise Exception('CausalNLP must be installed: pip install causalnlp')
-    from causalnlp.causalinference import CausalInferenceModel
+    from causalnlp import CausalInferenceModel
     return CausalInferenceModel(
                              df,
                              method=method,

ktrain/tests/test_ci.py

@@ -19,7 +19,7 @@ def adult_census():
 
 
 
-class TestMultilabel(TestCase):
+class TestCausalInference(TestCase):
 
     def test_multilabel(self):
         df  = adult_census()

ktrain/text/qa/core.py

@@ -67,6 +67,7 @@ def __init__(self, bert_squad_model='bert-large-uncased-whole-word-masking-finet
         try:
             self.model = TFAutoModelForQuestionAnswering.from_pretrained(self.model_name)
         except:
+            warnings.warn('Could not load supplied model as TensorFlow checkpoint - attempting to load using from_pt=True')
             self.model = TFAutoModelForQuestionAnswering.from_pretrained(self.model_name, from_pt=True)
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
         self.maxlen = 512

ktrain/version.py

@@ -1,2 +1,2 @@
 __all__ = ['__version__']
-__version__ = '0.27.2'
+__version__ = '0.27.3'