make get_widget_defaults private

src/app.py

@@ -1,7 +1,6 @@
 """The App module is the main entry point for the application.
 
     Run `streamlit run app.py` to start the app.
-
 """
 
 import pandas as pd
@@ -53,7 +52,7 @@ def _show_menu(pages: list[Page]) -> int:
 def _initialize_session_state(pages: list[Page]):
     if "active_page" not in st.session_state:
         for page in pages:
-            st.session_state.update(**page.get_widget_defaults())
+            st.session_state.update(**page._get_widget_defaults())
     st.session_state.update(st.session_state)
 
 

src/subpages/attention.py

@@ -91,7 +91,7 @@ class AttentionPage(Page):
     name = "Activations"
     icon = "activity"
 
-    def get_widget_defaults(self):
+    def _get_widget_defaults(self):
         return {
             "act_n_components": 8,
             "act_default_text": """Now I ask you: what can be expected of man since he is a being endowed with strange qualities? Shower upon him every earthly blessing, drown him in a sea of happiness, so that nothing but bubbles of bliss can be seen on the surface; give him economic prosperity, such that he should have nothing else to do but sleep, eat cakes and busy himself with the continuation of his species, and even then out of sheer ingratitude, sheer spite, man would play you some nasty trick. He would even risk his cakes and would deliberately desire the most fatal rubbish, the most uneconomical absurdity, simply to introduce into all this positive good sense his fatal fantastic element. It is just his fantastic dreams, his vulgar folly that he will desire to retain, simply in order to prove to himself--as though that were so necessary-- that men still are men and not the keys of a piano, which the laws of nature threaten to control so completely that soon one will be able to desire nothing but by the calendar. And that is not all: even if man really were nothing but a piano-key, even if this were proved to him by natural science and mathematics, even then he would not become reasonable, but would purposely do something perverse out of simple ingratitude, simply to gain his point. And if he does not find means he will contrive destruction and chaos, will contrive sufferings of all sorts, only to gain his point! He will launch a curse upon the world, and as only man can curse (it is his privilege, the primary distinction between him and other animals), may be by his curse alone he will attain his object--that is, convince himself that he is a man and not a piano-key!""",

src/subpages/find_duplicates.py

@@ -15,7 +15,7 @@ class FindDuplicatesPage(Page):
     name = "Find Duplicates"
     icon = "fingerprint"
 
-    def get_widget_defaults(self):
+    def _get_widget_defaults(self):
         return {
             "cutoff": 0.95,
         }

src/subpages/hidden_states.py

@@ -71,7 +71,7 @@ class HiddenStatesPage(Page):
     name = "Hidden States"
     icon = "grid-3x3"
 
-    def get_widget_defaults(self):
+    def _get_widget_defaults(self):
         return {
             "n_tokens": 1_000,
             "svd_n_iter": 5,

src/subpages/home.py

@@ -37,7 +37,7 @@ class HomePage(Page):
     name = "Home / Setup"
     icon = "house"
 
-    def get_widget_defaults(self):
+    def _get_widget_defaults(self):
         return {
             "encoder_model_name": _SENTENCE_ENCODER_MODEL,
             "model_name": _MODEL_NAME,

src/subpages/lossy_samples.py

@@ -15,7 +15,7 @@ class LossySamplesPage(Page):
     name = "Samples by Loss"
     icon = "sort-numeric-down-alt"
 
-    def get_widget_defaults(self):
+    def _get_widget_defaults(self):
         return {
             "skip_correct": True,
             "samples_by_loss_show_df": True,

src/subpages/metrics.py

@@ -52,7 +52,7 @@ class MetricsPage(Page):
     name = "Metrics"
     icon = "graph-up-arrow"
 
-    def get_widget_defaults(self):
+    def _get_widget_defaults(self):
         return {
             "normalize": True,
             "zero_diagonal": False,

src/subpages/page.py

@@ -10,7 +10,7 @@ from transformers import AutoTokenizer  # type: ignore
 
 @dataclass
 class Context:
-    """This object facilitates passing around the applications state between different pages."""
+    """This object facilitates passing around the application's state between different pages."""
 
     model: AutoModelForSequenceClassification
     tokenizer: AutoTokenizer
@@ -34,8 +34,8 @@ class Page:
     name: str
     icon: str
 
-    def get_widget_defaults(self):
-        """This function holds the default settings for all the page's widgets.
+    def _get_widget_defaults(self):
+        """This function holds the default settings for all widgets contained on this page.
 
         Returns:
             dict: A dictionary of widget defaults, where the keys are the widget names and the values are the default.

src/subpages/probing.py

@@ -29,7 +29,7 @@ class ProbingPage(Page):
     name = "Probing"
     icon = "fonts"
 
-    def get_widget_defaults(self):
+    def _get_widget_defaults(self):
         return {"probing_textarea": _DEFAULT_SENTENCES}
 
     def render(self, context: Context):

src/subpages/random_samples.py

@@ -10,7 +10,7 @@ class RandomSamplesPage(Page):
     name = "Random Samples"
     icon = "shuffle"
 
-    def get_widget_defaults(self):
+    def _get_widget_defaults(self):
         return {
             "random_sample_size_min": 128,
         }