updated setup.py

Gaius-Augustus · Sep 10, 2024 · 3514955 · 3514955
1 parent 46ca967
commit 3514955
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/setup.py b/setup.py
@@ -16,7 +16,6 @@
         include=["learnMSA", "learnMSA.run", "learnMSA.msa_hmm", "learnMSA.protein_language_models"]
     ),
     install_requires=["tensorflow>=2.5.0,<2.11",
-                      "tensorflow_probability<=0.18.0",
                       "networkx",
                       "logomaker", 
                       "seaborn",
@@ -27,10 +26,10 @@
                       "SentencePiece",
                       "numpy<2.0"],
     include_package_data=True,
-    package_data={'': ["msa_hmm/trained_prior/*/*", 
-                        "msa_hmm/trained_prior/transition_priors/*/*",
-                        "protein_language_models/new_scoring_models_frozen/*/*",
-                        "protein_language_models/priors_V3/*/*"]},
+    package_data={'': ["msa_hmm/trained_prior/*", 
+                        "msa_hmm/trained_prior/transition_priors/*",
+                        "protein_language_models/new_scoring_models_frozen/*",
+                        "protein_language_models/priors_V3/*"]},
     license="MIT",
     license_files = ("LICENSE.md"),
     entry_points={