def select_mode()

in obelics/visualization/web_document_and_filtering_visualization.py [0:0]

• 15 lines of code
• 5 McCabe index (conditional complexity)

    def select_mode(self):
        st.header("Select a mode")
        options_mode = ["All original web documents", "Retained web documents", "Discarded web documents"]
        self.mode = st.selectbox(label="Select a mode", options=options_mode)
        if self.mode == options_mode[0]:
            self.dataset = self.full_dataset
        if self.mode == options_mode[1]:
            self.dataset = st.session_state.retained_web_document_dataset
        if self.mode == options_mode[2]:
            self.dataset = st.session_state.discarded_web_document_dataset

        if self.dataset is None:
            st.warning(
                "To display retained or discarded documents, please perform first the filtering pipeline in the box"
                " above"
            )