Update

.pre-commit-config.yaml

@@ -8,7 +8,3 @@ repos:
     rev: 22.6.0
     hooks:
       - id: black-jupyter
-  - repo: https://github.com/kynan/nbstripout
-    rev: 0.5.0
-    hooks:
-      - id: nbstripout

app_lib/main.py

@@ -1,14 +1,14 @@
-import torch
 import streamlit as st
+import torch
 
+from app_lib.test import get_testing_config, test
 from app_lib.user_input import (
+    get_advanced_settings,
     get_class_name,
     get_concepts,
     get_image,
     get_model_name,
-    get_advanced_settings,
 )
-from app_lib.test import get_testing_config, test
 from app_lib.viz import viz_results
 
 
@@ -56,9 +56,13 @@ def main(device=torch.device("cuda" if torch.cuda.is_available() else "cpu")):
             st.error(error_message)
 
         with st.container():
-            significance_level, tau_max, r, cardinality, dataset_name = get_advanced_settings(
-                concepts, concepts_ready
-            )
+            (
+                significance_level,
+                tau_max,
+                r,
+                cardinality,
+                dataset_name,
+            ) = get_advanced_settings(concepts, concepts_ready)
 
             test_button = st.button(
                 "Test Concepts",

app_lib/test.py

@@ -1,17 +1,17 @@
-import torch
-import clip
-import open_clip
-import h5py
-import streamlit as st
-import numpy as np
 from concurrent.futures import ThreadPoolExecutor, as_completed
 
+import clip
+import h5py
 import ml_collections
+import numpy as np
+import open_clip
+import streamlit as st
+import torch
 from huggingface_hub import hf_hub_download
 
-from ibydmt.test import xSKIT
-from app_lib.utils import SUPPORTED_MODELS
 from app_lib.ckde import cKDE
+from app_lib.utils import SUPPORTED_MODELS
+from ibydmt.test import xSKIT
 
 rng = np.random.default_rng()
 

app_lib/user_input.py

@@ -2,7 +2,7 @@ import streamlit as st
 from PIL import Image
 from streamlit_image_select import image_select
 
-from app_lib.utils import SUPPORTED_MODELS, SUPPORTED_DATASETS
+from app_lib.utils import SUPPORTED_DATASETS, SUPPORTED_MODELS
 
 
 def _validate_class_name(class_name):

app_lib/viz.py

@@ -1,6 +1,6 @@
-import streamlit as st
 import pandas as pd
 import plotly.express as px
+import streamlit as st
 
 
 def _viz_wealth(results):