Blacken code

edit.py

@@ -36,13 +36,20 @@ from skops import card
 from skops.card._model_card import PlotSection, split_subsection_names
 
 from utils import iterate_key_section_content, process_card_for_rendering
-from tasks import AddSectionTask, AddFigureTask, DeleteSectionTask, TaskState, UpdateFigureTask, UpdateSectionTask
+from tasks import (
+    AddSectionTask,
+    AddFigureTask,
+    DeleteSectionTask,
+    TaskState,
+    UpdateFigureTask,
+    UpdateSectionTask,
+)
 
 
 arepr = reprlib.Repr()
 arepr.maxstring = 24
 tmp_path = Path(mkdtemp(prefix="skops-"))  # temporary files
-hf_path = Path(mkdtemp(prefix="skops-"))   # hf repo
+hf_path = Path(mkdtemp(prefix="skops-"))  # hf repo
 
 
 def load_model_card_from_repo(repo_id: str) -> card.Card:
@@ -53,7 +60,11 @@ def load_model_card_from_repo(repo_id: str) -> card.Card:
 
 
 def _update_model_card(
-    model_card: card.Card, key: str, section_name: str, content: str, is_fig: bool,
+    model_card: card.Card,
+    key: str,
+    section_name: str,
+    content: str,
+    is_fig: bool,
 ) -> None:
     # This is a very roundabout way to update the model card but it's necessary
     # because of how streamlit handles session state. Basically, there have to
@@ -109,7 +120,9 @@ def _update_model_card(
 
 def _add_section(model_card: card.Card, key: str) -> None:
     section_name = f"{key}/Untitled"
-    task = AddSectionTask(model_card, title=section_name, content="[More Information Needed]")
+    task = AddSectionTask(
+        model_card, title=section_name, content="[More Information Needed]"
+    )
     st.session_state.task_state.add(task)
 
 
@@ -157,7 +170,11 @@ def _add_fig_form(
 
 
 def create_form_from_section(
-    model_card: card.Card, key: str, section_name: str, content: str, is_fig: bool = False
+    model_card: card.Card,
+    key: str,
+    section_name: str,
+    content: str,
+    is_fig: bool = False,
 ) -> None:
     split_sections = split_subsection_names(section_name)
     old_title = split_sections[-1]
@@ -203,7 +220,9 @@ def create_form_from_section(
 
 
 def display_sections(model_card: card.Card) -> None:
-    for key, section_name, content, is_fig in iterate_key_section_content(model_card._data):
+    for key, section_name, content, is_fig in iterate_key_section_content(
+        model_card._data
+    ):
         create_form_from_section(model_card, key, section_name, content, is_fig)
 
 
@@ -248,9 +267,7 @@ def add_download_model_card_button():
     model_card = st.session_state.get("model_card")
     download_disabled = not bool(model_card)
     data = model_card.render()
-    st.download_button(
-        "Save (md)", data=data, disabled=download_disabled
-    )
+    st.download_button("Save (md)", data=data, disabled=download_disabled)
 
 
 def edit_input_form():

start.py

@@ -29,7 +29,7 @@ import skops.io as sio
 from skops import card, hub_utils
 
 
-hf_path = Path(mkdtemp(prefix="skops-"))   # hf repo
+hf_path = Path(mkdtemp(prefix="skops-"))  # hf repo
 tmp_path = Path(mkdtemp(prefix="skops-"))  # temporary files
 description = """Create an sklearn model card
 
@@ -112,7 +112,9 @@ def create_skops_model_card() -> None:
 def create_empty_model_card() -> None:
     init_repo(hf_path)
     metadata = card.metadata_from_config(hf_path)
-    model_card = card.Card(model=st.session_state.model, metadata=metadata, template=None)
+    model_card = card.Card(
+        model=st.session_state.model, metadata=metadata, template=None
+    )
     model_card.add(**{"Untitled": "[More Information Needed]"})
     st.session_state.model_card = model_card
 
@@ -167,7 +169,7 @@ def start_input_form():
         ],
         key="task",
         on_change=init_repo,
-        args=(hf_path,)
+        args=(hf_path,),
     )
     st.markdown("---")
 
@@ -176,7 +178,7 @@ def start_input_form():
         value=f"scikit-learn=={sklearn.__version__}\n",
         key="requirements",
         on_change=init_repo,
-        args=(hf_path,)
+        args=(hf_path,),
     )
     st.markdown("---")
 

tasks.py

@@ -15,6 +15,7 @@ from streamlit.runtime.uploaded_file_manager import UploadedFile
 
 class Task:
     """(Abstract) base class for tasks"""
+
     def do(self) -> None:
         raise NotImplementedError
 
@@ -24,6 +25,7 @@ class Task:
 
 class TaskState:
     """Tracking the state of tasks"""
+
     def __init__(self) -> None:
         self.done_list: list[Task] = []
         self.undone_list: list[Task] = []
@@ -56,6 +58,7 @@ class TaskState:
 
 class AddSectionTask(Task):
     """Add a new text section"""
+
     def __init__(
         self,
         model_card: card.Card,
@@ -78,6 +81,7 @@ class AddSectionTask(Task):
 
 class AddFigureTask(Task):
     """Add a new figure section"""
+
     def __init__(
         self,
         model_card: card.Card,
@@ -106,6 +110,7 @@ class DeleteSectionTask(Task):
     but only turned invisible.
 
     """
+
     def __init__(
         self,
         model_card: card.Card,
@@ -123,6 +128,7 @@ class DeleteSectionTask(Task):
 
 class UpdateSectionTask(Task):
     """Change the title or content of a text section"""
+
     def __init__(
         self,
         model_card: card.Card,
@@ -154,6 +160,7 @@ class UpdateSectionTask(Task):
 
 class UpdateFigureTask(Task):
     """Change the title or image of a figure section"""
+
     def __init__(
         self,
         model_card: card.Card,