push required python to 3.9

imodelsx/augtree/embed.py

@@ -11,7 +11,7 @@
 from os.path import join
 import pickle as pkl
 import sklearn.metrics
-import fire
+# import fire
 import torch.cuda
 from scipy.spatial import distance
 # from numba import jit
@@ -169,10 +169,10 @@ def test_dists():
     assert np.allclose(dists_eucl, dists_eucl_ref)
 
 
-if __name__ == '__main__':
+# if __name__ == '__main__':
     # test_dists()
     # allows calling wth args, e.g. python embed.py --dataset_name sst2
-    fire.Fire(EmbsManager)
+    # fire.Fire(EmbsManager)
 
     # embs = EmbsManager(
     #     dataset_name='financial_phrasebank',

imodelsx/augtree/utils.py

@@ -2,7 +2,6 @@
 import numpy as np
 from spacy.lang.en import English
 from sklearn.metrics import mean_squared_error
-from nltk.stem.porter import PorterStemmer
 ROOT = 0
 LEFT = 1
 RIGHT = 2
@@ -95,6 +94,7 @@ def __init__(self, convert_lower, use_stemming):
         self.convert_lower = convert_lower
         self.use_stemming = use_stemming
         if self.use_stemming:
+            from nltk.stem.porter import PorterStemmer
             self.stemmer = PorterStemmer()
 
     def __call__(self, s):

readme.md

@@ -8,7 +8,7 @@
 </p>
 <p align="center">
   <img src="https://img.shields.io/badge/license-mit-blue.svg">
-  <img src="https://img.shields.io/badge/python-3.6+-blue">
+  <img src="https://img.shields.io/badge/python-3.9+-blue">
   <img src="https://img.shields.io/pypi/v/imodelsx?color=green">  
 </p>  
 

setup.py

@@ -35,7 +35,7 @@
         exclude=['tests', 'tests.*', '*.test.*']
     ),
     install_requires=required_pypi,
-    python_requires='>=3.6',
+    python_requires='>=3.9',
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: MIT License",