updated import, requirements

beetles_tda/features.py

@@ -1,4 +1,4 @@
-import giotto.diagrams as diag
+import gtda.diagrams as diag
 import numpy as np
 import pandas as pd
 

beetles_tda/generate_data.py

@@ -6,16 +6,16 @@
 from sklearn.model_selection import train_test_split
 from sklearn.cluster import KMeans
 from sklearn.metrics.cluster import homogeneity_score
-import giotto.time_series as ts
-import giotto.homology as hl
-from giotto.pipeline import Pipeline
+import gtda.time_series as ts
+import gtda.homology as hl
+from gtda.pipeline import Pipeline
 from beetles_tda.features import (
     get_amplitude,
     get_max_lifetime,
     get_mean_lifetime,
     get_n_rel_holes,
 )
-import giotto.diagrams as diag
+import gtda.diagrams as diag
 from fastprogress import master_bar, progress_bar
 
 

beetles_tda/visualize.py

@@ -1,6 +1,6 @@
 import numpy as np
 import plotly.graph_objs as gobj
-from giotto.diagrams._utils import _subdiagrams
+from gtda.diagrams._utils import _subdiagrams
 import seaborn as sns
 import matplotlib.pyplot as plt
 

requirements.txt

@@ -1,11 +1,12 @@
-giotto-learn==0.1.3
+giotto-tda==0.1.4
 jupyterlab==1.2.3
 matplotlib==3.1.2
 numpy==1.17.4
 pandas==0.25.3
-scikit-learn==0.21.3
+cython==0.29.15
+scikit-learn==0.22
 seaborn==0.9.0
 tqdm==4.40.0
-plotly==4.3.0
+plotly==4.4.1
 fire==0.2.1
-fastprogress==0.1.22
+fastprogress==0.1.22