Downgrade streamlit - 1.17

hexviz/attention.py

@@ -67,7 +67,7 @@ def get_tape_bert() -> Tuple[TAPETokenizer, ProteinBertModel]:
     model = ProteinBertModel.from_pretrained('bert-base', output_attentions=True)
     return tokenizer, model
 
-@st.cache_data
+@st.cache
 def get_attention(
     sequence: List[str], model_type: ModelType = ModelType.TAPE_BERT  
 ):
@@ -105,7 +105,7 @@ def unidirectional_sum_filtered(attention, layer, head, threshold):
                 unidirectional_sum_for_head.append((sum, i, j))
     return unidirectional_sum_for_head
 
-@st.cache_data
+@st.cache
 def get_attention_pairs(pdb_code: str, layer: int, head: int, threshold: int = 0.2, model_type: ModelType = ModelType.TAPE_BERT):
     # fetch structure
     structure = get_structure(pdb_code=pdb_code)

poetry.lock

@@ -1686,14 +1686,14 @@ streamlit = ">=0.63"
 
 [[package]]
 name = "streamlit"
-version = "1.20.0"
+version = "1.17.0"
 description = "The fastest way to build data apps in Python"
 category = "main"
 optional = false
 python-versions = ">=3.7, !=3.9.7"
 
 [package.dependencies]
-altair = ">=3.2.0,<5"
+altair = ">=3.2.0"
 blinker = ">=1.0.0"
 cachetools = ">=4.0"
 click = ">=7.0"
@@ -1701,7 +1701,7 @@ gitpython = "!=3.1.19"
 importlib-metadata = ">=1.4"
 numpy = "*"
 packaging = ">=14.1"
-pandas = ">=0.25,<2"
+pandas = ">=0.21.0"
 pillow = ">=6.2.0"
 protobuf = ">=3.12,<4"
 pyarrow = ">=4.0"
@@ -1712,7 +1712,7 @@ requests = ">=2.4"
 rich = ">=10.11.0"
 semver = "*"
 toml = "*"
-tornado = ">=6.0.3"
+tornado = ">=5.0"
 typing-extensions = ">=3.10.0.0"
 tzlocal = ">=1.1"
 validators = ">=0.2"
@@ -2133,8 +2133,8 @@ testing = ["pytest (>=6)", "pytest-checkdocs (>=2.4)", "flake8 (<5)", "pytest-co
 
 [metadata]
 lock-version = "1.1"
-python-versions = "^3.8.9"
-content-hash = "c4902982dd427349993b8019f7130f81583d025190b4bd2a2452166c7d146249"
+python-versions = ">=3.8.9,<3.9.7 || >3.9.7,<4.0.0"
+content-hash = "f7215288fc64984519d9a9ca7b7db5663b34c7727322547d6c8e9e87fdd1161b"
 
 [metadata.files]
 altair = []

pyproject.toml

@@ -5,8 +5,8 @@ description = "Visualize and analyze attention patterns for protein language mod
 authors = ["Aksel Lenes <[email protected]>"]
 
 [tool.poetry.dependencies]
-python = "^3.8.9"
-streamlit = "^1.20.0"
+python = ">=3.8.9,<3.9.7 || >3.9.7,<4.0.0"
+streamlit = "1.17"
 stmol = "^0.0.9"
 biopython = "^1.81"
 transformers = "^4.27.1"