edsl 0.1.33__py3-none-any.whl → 0.1.33.dev1__py3-none-any.whl

edsl/scenarios/ScenarioList.py

@@ -6,11 +6,6 @@ import csv
 import random
 from collections import UserList, Counter
 from collections.abc import Iterable
-import urllib.parse
-import urllib.request
-from io import StringIO
-from collections import defaultdict
-import inspect
 
 from simpleeval import EvalWithCompoundTypes
 
@@ -20,9 +15,6 @@ from edsl.scenarios.Scenario import Scenario
 from edsl.scenarios.ScenarioListPdfMixin import ScenarioListPdfMixin
 from edsl.scenarios.ScenarioListExportMixin import ScenarioListExportMixin
 
-from edsl.conjure.naming_utilities import sanitize_string
-from edsl.utilities.utilities import is_valid_variable_name
-
 
 class ScenarioListMixin(ScenarioListPdfMixin, ScenarioListExportMixin):
     pass
@@ -31,180 +23,12 @@ class ScenarioListMixin(ScenarioListPdfMixin, ScenarioListExportMixin):
 class ScenarioList(Base, UserList, ScenarioListMixin):
     """Class for creating a list of scenarios to be used in a survey."""
 
-    def __init__(self, data: Optional[list] = None, codebook: Optional[dict] = None):
+    def __init__(self, data: Optional[list] = None):
         """Initialize the ScenarioList class."""
         if data is not None:
             super().__init__(data)
         else:
             super().__init__([])
-        self.codebook = codebook or {}
-
-    @property
-    def has_jinja_braces(self) -> bool:
-        """Check if the ScenarioList has Jinja braces."""
-        return any([scenario.has_jinja_braces for scenario in self])
-
-    def convert_jinja_braces(self) -> ScenarioList:
-        """Convert Jinja braces to Python braces."""
-        return ScenarioList([scenario.convert_jinja_braces() for scenario in self])
-
-    def give_valid_names(self) -> ScenarioList:
-        """Give valid names to the scenario keys.
-
-        >>> s = ScenarioList([Scenario({'a': 1, 'b': 2}), Scenario({'a': 1, 'b': 1})])
-        >>> s.give_valid_names()
-        ScenarioList([Scenario({'a': 1, 'b': 2}), Scenario({'a': 1, 'b': 1})])
-        >>> s = ScenarioList([Scenario({'are you there John?': 1, 'b': 2}), Scenario({'a': 1, 'b': 1})])
-        >>> s.give_valid_names()
-        ScenarioList([Scenario({'john': 1, 'b': 2}), Scenario({'a': 1, 'b': 1})])
-        """
-        codebook = {}
-        new_scenaerios = []
-        for scenario in self:
-            new_scenario = {}
-            for key in scenario:
-                if not is_valid_variable_name(key):
-                    if key in codebook:
-                        new_key = codebook[key]
-                    else:
-                        new_key = sanitize_string(key)
-                        if not is_valid_variable_name(new_key):
-                            new_key = f"var_{len(codebook)}"
-                        codebook[key] = new_key
-                    new_scenario[new_key] = scenario[key]
-                else:
-                    new_scenario[key] = scenario[key]
-            new_scenaerios.append(Scenario(new_scenario))
-        return ScenarioList(new_scenaerios, codebook)
-
-    def unpivot(self, id_vars=None, value_vars=None):
-        """
-        Unpivot the ScenarioList, allowing for id variables to be specified.
-
-        Parameters:
-        id_vars (list): Fields to use as identifier variables (kept in each entry)
-        value_vars (list): Fields to unpivot. If None, all fields not in id_vars will be used.
-
-        Example:
-        >>> s = ScenarioList([
-        ...     Scenario({'id': 1, 'year': 2020, 'a': 10, 'b': 20}),
-        ...     Scenario({'id': 2, 'year': 2021, 'a': 15, 'b': 25})
-        ... ])
-        >>> s.unpivot(id_vars=['id', 'year'], value_vars=['a', 'b'])
-        ScenarioList([Scenario({'id': 1, 'year': 2020, 'variable': 'a', 'value': 10}), Scenario({'id': 1, 'year': 2020, 'variable': 'b', 'value': 20}), Scenario({'id': 2, 'year': 2021, 'variable': 'a', 'value': 15}), Scenario({'id': 2, 'year': 2021, 'variable': 'b', 'value': 25})])
-        """
-        if id_vars is None:
-            id_vars = []
-        if value_vars is None:
-            value_vars = [field for field in self[0].keys() if field not in id_vars]
-
-        new_scenarios = []
-        for scenario in self:
-            for var in value_vars:
-                new_scenario = {id_var: scenario[id_var] for id_var in id_vars}
-                new_scenario["variable"] = var
-                new_scenario["value"] = scenario[var]
-                new_scenarios.append(Scenario(new_scenario))
-
-        return ScenarioList(new_scenarios)
-
-    def pivot(self, id_vars, var_name="variable", value_name="value"):
-        """
-        Pivot the ScenarioList from long to wide format.
-
-        Parameters:
-        id_vars (list): Fields to use as identifier variables
-        var_name (str): Name of the variable column (default: 'variable')
-        value_name (str): Name of the value column (default: 'value')
-
-        Example:
-        >>> s = ScenarioList([
-        ...     Scenario({'id': 1, 'year': 2020, 'variable': 'a', 'value': 10}),
-        ...     Scenario({'id': 1, 'year': 2020, 'variable': 'b', 'value': 20}),
-        ...     Scenario({'id': 2, 'year': 2021, 'variable': 'a', 'value': 15}),
-        ...     Scenario({'id': 2, 'year': 2021, 'variable': 'b', 'value': 25})
-        ... ])
-        >>> s.pivot(id_vars=['id', 'year'])
-        ScenarioList([Scenario({'id': 1, 'year': 2020, 'a': 10, 'b': 20}), Scenario({'id': 2, 'year': 2021, 'a': 15, 'b': 25})])
-        """
-        pivoted_dict = {}
-
-        for scenario in self:
-            # Create a tuple of id values to use as a key
-            id_key = tuple(scenario[id_var] for id_var in id_vars)
-
-            # If this combination of id values hasn't been seen before, initialize it
-            if id_key not in pivoted_dict:
-                pivoted_dict[id_key] = {id_var: scenario[id_var] for id_var in id_vars}
-
-            # Add the variable-value pair to the dict
-            variable = scenario[var_name]
-            value = scenario[value_name]
-            pivoted_dict[id_key][variable] = value
-
-        # Convert the dict of dicts to a list of Scenarios
-        pivoted_scenarios = [
-            Scenario(dict(zip(id_vars, id_key), **values))
-            for id_key, values in pivoted_dict.items()
-        ]
-
-        return ScenarioList(pivoted_scenarios)
-
-    def group_by(self, id_vars, variables, func):
-        """
-        Group the ScenarioList by id_vars and apply a function to the specified variables.
-
-        Parameters:
-        id_vars (list): Fields to use as identifier variables for grouping
-        variables (list): Fields to pass to the aggregation function
-        func (callable): Function to apply to the grouped variables.
-                        Should accept lists of values for each variable.
-
-        Returns:
-        ScenarioList: A new ScenarioList with the grouped and aggregated results
-
-        Example:
-        >>> def avg_sum(a, b):
-        ...     return {'avg_a': sum(a) / len(a), 'sum_b': sum(b)}
-        >>> s = ScenarioList([
-        ...     Scenario({'group': 'A', 'year': 2020, 'a': 10, 'b': 20}),
-        ...     Scenario({'group': 'A', 'year': 2021, 'a': 15, 'b': 25}),
-        ...     Scenario({'group': 'B', 'year': 2020, 'a': 12, 'b': 22}),
-        ...     Scenario({'group': 'B', 'year': 2021, 'a': 17, 'b': 27})
-        ... ])
-        >>> s.group_by(id_vars=['group'], variables=['a', 'b'], func=avg_sum)
-        ScenarioList([Scenario({'group': 'A', 'avg_a': 12.5, 'sum_b': 45}), Scenario({'group': 'B', 'avg_a': 14.5, 'sum_b': 49})])
-        """
-        # Check if the function is compatible with the specified variables
-        func_params = inspect.signature(func).parameters
-        if len(func_params) != len(variables):
-            raise ValueError(
-                f"Function {func.__name__} expects {len(func_params)} arguments, but {len(variables)} variables were provided"
-            )
-
-        # Group the scenarios
-        grouped = defaultdict(lambda: defaultdict(list))
-        for scenario in self:
-            key = tuple(scenario[id_var] for id_var in id_vars)
-            for var in variables:
-                grouped[key][var].append(scenario[var])
-
-        # Apply the function to each group
-        result = []
-        for key, group in grouped.items():
-            try:
-                aggregated = func(*[group[var] for var in variables])
-            except Exception as e:
-                raise ValueError(f"Error applying function to group {key}: {str(e)}")
-
-            if not isinstance(aggregated, dict):
-                raise ValueError(f"Function {func.__name__} must return a dictionary")
-
-            new_scenario = dict(zip(id_vars, key))
-            new_scenario.update(aggregated)
-            result.append(Scenario(new_scenario))
-
-        return ScenarioList(result)
 
     @property
     def parameters(self) -> set:
@@ -282,10 +106,6 @@ class ScenarioList(Base, UserList, ScenarioListMixin):
         for s in data["scenarios"]:
             _ = s.pop("edsl_version")
             _ = s.pop("edsl_class_name")
-        for scenario in data["scenarios"]:
-            for key, value in scenario.items():
-                if hasattr(value, "to_dict"):
-                    data[key] = value.to_dict()
         return data_to_html(data)
 
     def tally(self, field) -> dict:
@@ -333,71 +153,6 @@ class ScenarioList(Base, UserList, ScenarioListMixin):
                 new_scenarios.append(new_scenario)
         return ScenarioList(new_scenarios)
 
-    def concatenate(self, fields: List[str], separator: str = ";") -> "ScenarioList":
-        """Concatenate specified fields into a single field.
-
-        Args:
-            fields (List[str]): List of field names to concatenate.
-            separator (str, optional): Separator to use between field values. Defaults to ";".
-
-        Returns:
-            ScenarioList: A new ScenarioList with concatenated fields.
-
-        Example:
-            >>> s = ScenarioList([Scenario({'a': 1, 'b': 2, 'c': 3}), Scenario({'a': 4, 'b': 5, 'c': 6})])
-            >>> s.concatenate(['a', 'b', 'c'])
-            ScenarioList([Scenario({'concat_a_b_c': '1;2;3'}), Scenario({'concat_a_b_c': '4;5;6'})])
-        """
-        new_scenarios = []
-        for scenario in self:
-            new_scenario = scenario.copy()
-            concat_values = []
-            for field in fields:
-                if field in new_scenario:
-                    concat_values.append(str(new_scenario[field]))
-                    del new_scenario[field]
-
-            new_field_name = f"concat_{'_'.join(fields)}"
-            new_scenario[new_field_name] = separator.join(concat_values)
-            new_scenarios.append(new_scenario)
-
-        return ScenarioList(new_scenarios)
-
-    def unpack_dict(
-        self, field: str, prefix: Optional[str] = None, drop_field: bool = False
-    ) -> ScenarioList:
-        """Unpack a dictionary field into separate fields.
-
-        Example:
-
-        >>> s = ScenarioList([Scenario({'a': 1, 'b': {'c': 2, 'd': 3}})])
-        >>> s.unpack_dict('b')
-        ScenarioList([Scenario({'a': 1, 'b': {'c': 2, 'd': 3}, 'c': 2, 'd': 3})])
-        """
-        new_scenarios = []
-        for scenario in self:
-            new_scenario = scenario.copy()
-            for key, value in scenario[field].items():
-                if prefix:
-                    new_scenario[prefix + key] = value
-                else:
-                    new_scenario[key] = value
-            if drop_field:
-                new_scenario.pop(field)
-            new_scenarios.append(new_scenario)
-        return ScenarioList(new_scenarios)
-
-    def transform(
-        self, field: str, func: Callable, new_name: Optional[str] = None
-    ) -> ScenarioList:
-        """Transform a field using a function."""
-        new_scenarios = []
-        for scenario in self:
-            new_scenario = scenario.copy()
-            new_scenario[new_name or field] = func(scenario[field])
-            new_scenarios.append(new_scenario)
-        return ScenarioList(new_scenarios)
-
     def mutate(
         self, new_var_string: str, functions_dict: Optional[dict[str, Callable]] = None
     ) -> ScenarioList:
@@ -555,19 +310,6 @@ class ScenarioList(Base, UserList, ScenarioListMixin):
         data = [{key: [scenario[key] for scenario in self.data]} for key in keys]
         return Dataset(data)
 
-    def split(
-        self, field: str, split_on: str, index: int, new_name: Optional[str] = None
-    ) -> ScenarioList:
-        """Split a scenario fiel in multiple fields."""
-        if new_name is None:
-            new_name = field + "_split_" + str(index)
-        new_scenarios = []
-        for scenario in self:
-            new_scenario = scenario.copy()
-            new_scenario[new_name] = scenario[field].split(split_on)[index]
-            new_scenarios.append(new_scenario)
-        return ScenarioList(new_scenarios)
-
     def add_list(self, name, values) -> ScenarioList:
         """Add a list of values to a ScenarioList.
 
@@ -650,6 +392,37 @@ class ScenarioList(Base, UserList, ScenarioListMixin):
 
         return ScenarioList([Scenario(entry) for entry in processed_lines])
 
+    @classmethod
+    def from_docx(cls, docx_file_path: str):
+        from docx import Document
+
+        doc = Document(docx_file_path)
+        lines = []
+
+        # Extract text from paragraphs, treating each paragraph as a line
+        for para in doc.paragraphs:
+            lines.extend(para.text.splitlines())
+
+        processed_lines = []
+        non_blank_lines = [
+            (i, line.strip()) for i, line in enumerate(lines) if line.strip()
+        ]
+
+        for index, (line_no, text) in enumerate(non_blank_lines):
+            entry = {
+                "line_no": line_no + 1,  # Using 1-based index for line numbers
+                "text": text,
+                "line_before": non_blank_lines[index - 1][1] if index > 0 else None,
+                "line_after": (
+                    non_blank_lines[index + 1][1]
+                    if index < len(non_blank_lines) - 1
+                    else None
+                ),
+            }
+            processed_lines.append(entry)
+
+        return ScenarioList([Scenario(entry) for entry in processed_lines])
+
     @classmethod
     def from_google_doc(cls, url: str) -> ScenarioList:
         """Create a ScenarioList from a Google Doc.
@@ -700,62 +473,6 @@ class ScenarioList(Base, UserList, ScenarioListMixin):
         """
         return cls([Scenario(row) for row in df.to_dict(orient="records")])
 
-    @classmethod
-    def from_wikipedia(cls, url: str, table_index: int = 0):
-        """
-        Extracts a table from a Wikipedia page.
-
-        Parameters:
-            url (str): The URL of the Wikipedia page.
-            table_index (int): The index of the table to extract (default is 0).
-
-        Returns:
-            pd.DataFrame: A DataFrame containing the extracted table.
-        # # Example usage
-        # url = "https://en.wikipedia.org/wiki/List_of_countries_by_GDP_(nominal)"
-        # df = from_wikipedia(url, 0)
-
-        # if not df.empty:
-        #     print(df.head())
-        # else:
-        #     print("Failed to extract table.")
-
-
-        """
-        import pandas as pd
-        import requests
-        from requests.exceptions import RequestException
-
-        try:
-            # Check if the URL is reachable
-            response = requests.get(url)
-            response.raise_for_status()  # Raises HTTPError for bad responses
-
-            # Extract tables from the Wikipedia page
-            tables = pd.read_html(url)
-
-            # Ensure the requested table index is within the range of available tables
-            if table_index >= len(tables) or table_index < 0:
-                raise IndexError(
-                    f"Table index {table_index} is out of range. This page has {len(tables)} table(s)."
-                )
-
-            # Return the requested table as a DataFrame
-            # return tables[table_index]
-            return cls.from_pandas(tables[table_index])
-
-        except RequestException as e:
-            print(f"Error fetching the URL: {e}")
-        except ValueError as e:
-            print(f"Error parsing tables: {e}")
-        except IndexError as e:
-            print(e)
-        except Exception as e:
-            print(f"An unexpected error occurred: {e}")
-
-        # Return an empty DataFrame in case of an error
-        # return cls.from_pandas(pd.DataFrame())
-
     def to_key_value(self, field: str, value=None) -> Union[dict, set]:
         """Return the set of values in the field.
 
@@ -877,15 +594,8 @@ class ScenarioList(Base, UserList, ScenarioListMixin):
         return cls.from_excel(temp_filename, sheet_name=sheet_name)
 
     @classmethod
-    def from_csv(cls, source: Union[str, urllib.parse.ParseResult]) -> ScenarioList:
-        """Create a ScenarioList from a CSV file or URL.
-
-        Args:
-            source: A string representing either a local file path or a URL to a CSV file,
-                    or a urllib.parse.ParseResult object for a URL.
-
-        Returns:
-            ScenarioList: A ScenarioList object containing the data from the CSV.
+    def from_csv(cls, filename: str) -> ScenarioList:
+        """Create a ScenarioList from a CSV file.
 
         Example:
 
@@ -901,37 +611,15 @@ class ScenarioList(Base, UserList, ScenarioListMixin):
         'Alice'
         >>> scenario_list[1]['age']
         '25'
-
-        >>> url = "https://example.com/data.csv"
-        >>> ## scenario_list_from_url = ScenarioList.from_csv(url)
         """
         from edsl.scenarios.Scenario import Scenario
 
-        def is_url(source):
-            try:
-                result = urllib.parse.urlparse(source)
-                return all([result.scheme, result.netloc])
-            except ValueError:
-                return False
-
-        if isinstance(source, str) and is_url(source):
-            with urllib.request.urlopen(source) as response:
-                csv_content = response.read().decode("utf-8")
-            csv_file = StringIO(csv_content)
-        elif isinstance(source, urllib.parse.ParseResult):
-            with urllib.request.urlopen(source.geturl()) as response:
-                csv_content = response.read().decode("utf-8")
-            csv_file = StringIO(csv_content)
-        else:
-            csv_file = open(source, "r")
-
-        try:
-            reader = csv.reader(csv_file)
+        observations = []
+        with open(filename, "r") as f:
+            reader = csv.reader(f)
             header = next(reader)
-            observations = [Scenario(dict(zip(header, row))) for row in reader]
-        finally:
-            csv_file.close()
-
+            for row in reader:
+                observations.append(Scenario(dict(zip(header, row))))
         return cls(observations)
 
     def _to_dict(self, sort=False) -> dict:
@@ -975,16 +663,6 @@ class ScenarioList(Base, UserList, ScenarioListMixin):
 
         return cls([Scenario.from_dict(s) for s in data["scenarios"]])
 
-    @classmethod
-    def from_nested_dict(cls, data: dict) -> ScenarioList:
-        """Create a `ScenarioList` from a nested dictionary."""
-        from edsl.scenarios.Scenario import Scenario
-
-        s = ScenarioList()
-        for key, value in data.items():
-            s.add_list(key, value)
-        return s
-
     def code(self) -> str:
         ## TODO: Refactor to only use the questions actually in the survey
         """Create the Python code representation of a survey."""

edsl/scenarios/ScenarioListExportMixin.py

@@ -41,12 +41,3 @@ class ScenarioListExportMixin(DatasetExportMixin):
     def __init_subclass__(cls, **kwargs):
         super().__init_subclass__(**kwargs)
         decorate_methods_from_mixin(cls, DatasetExportMixin)
-
-    def to_docx(self, filename: str):
-        """Export the ScenarioList to a .docx file."""
-        dataset = self.to_dataset()
-        from edsl.results.DatasetTree import Tree
-
-        tree = Tree(dataset)
-        tree.construct_tree()
-        tree.to_docx(filename)

edsl/scenarios/ScenarioListPdfMixin.py

@@ -1,161 +1,15 @@
 import fitz  # PyMuPDF
 import os
-import copy
 import subprocess
-import requests
-import tempfile
-import os
-
-# import urllib.parse as urlparse
-from urllib.parse import urlparse
 
 # from edsl import Scenario
 
-import requests
-import re
-import tempfile
-import os
-import atexit
-from urllib.parse import urlparse, parse_qs
-
-
-class GoogleDriveDownloader:
-    _temp_dir = None
-    _temp_file_path = None
-
-    @classmethod
-    def fetch_from_drive(cls, url, filename=None):
-        # Extract file ID from the URL
-        file_id = cls._extract_file_id(url)
-        if not file_id:
-            raise ValueError("Invalid Google Drive URL")
-
-        # Construct the download URL
-        download_url = f"https://drive.google.com/uc?export=download&id={file_id}"
-
-        # Send a GET request to the URL
-        session = requests.Session()
-        response = session.get(download_url, stream=True)
-        response.raise_for_status()
-
-        # Check for large file download prompt
-        for key, value in response.cookies.items():
-            if key.startswith("download_warning"):
-                params = {"id": file_id, "confirm": value}
-                response = session.get(download_url, params=params, stream=True)
-                break
-
-        # Create a temporary file to save the download
-        if not filename:
-            filename = "downloaded_file"
-
-        if cls._temp_dir is None:
-            cls._temp_dir = tempfile.TemporaryDirectory()
-            atexit.register(cls._cleanup)
-
-        cls._temp_file_path = os.path.join(cls._temp_dir.name, filename)
-
-        # Write the content to the temporary file
-        with open(cls._temp_file_path, "wb") as f:
-            for chunk in response.iter_content(32768):
-                if chunk:
-                    f.write(chunk)
-
-        print(f"File saved to: {cls._temp_file_path}")
-
-        return cls._temp_file_path
-
-    @staticmethod
-    def _extract_file_id(url):
-        # Try to extract file ID from '/file/d/' format
-        file_id_match = re.search(r"/d/([a-zA-Z0-9-_]+)", url)
-        if file_id_match:
-            return file_id_match.group(1)
-
-        # If not found, try to extract from 'open?id=' format
-        parsed_url = urlparse(url)
-        query_params = parse_qs(parsed_url.query)
-        if "id" in query_params:
-            return query_params["id"][0]
-
-        return None
-
-    @classmethod
-    def _cleanup(cls):
-        if cls._temp_dir:
-            cls._temp_dir.cleanup()
-
-    @classmethod
-    def get_temp_file_path(cls):
-        return cls._temp_file_path
-
-
-def fetch_and_save_pdf(url, filename):
-    # Send a GET request to the URL
-    response = requests.get(url)
-
-    # Check if the request was successful
-    response.raise_for_status()
-
-    # Create a temporary directory
-    with tempfile.TemporaryDirectory() as temp_dir:
-        # Construct the full path for the file
-        temp_file_path = os.path.join(temp_dir, filename)
-
-        # Write the content to the temporary file
-        with open(temp_file_path, "wb") as file:
-            file.write(response.content)
-
-        print(f"PDF saved to: {temp_file_path}")
-
-        # Here you can perform operations with the file
-        # The file will be automatically deleted when you exit this block
-
-    return temp_file_path
-
-
-# Example usage:
-# url = "https://example.com/sample.pdf"
-# fetch_and_save_pdf(url, "sample.pdf")
-
 
 class ScenarioListPdfMixin:
     @classmethod
-    def from_pdf(cls, filename_or_url, collapse_pages=False):
-        # Check if the input is a URL
-        if cls.is_url(filename_or_url):
-            # Check if it's a Google Drive URL
-            if "drive.google.com" in filename_or_url:
-                temp_filename = GoogleDriveDownloader.fetch_from_drive(
-                    filename_or_url, "temp_pdf.pdf"
-                )
-            else:
-                # For other URLs, use the previous fetch_and_save_pdf function
-                temp_filename = fetch_and_save_pdf(filename_or_url, "temp_pdf.pdf")
-
-            scenarios = list(cls.extract_text_from_pdf(temp_filename))
-        else:
-            # If it's not a URL, assume it's a local file path
-            scenarios = list(cls.extract_text_from_pdf(filename_or_url))
-        if not collapse_pages:
-            return cls(scenarios)
-        else:
-            txt = ""
-            for scenario in scenarios:
-                txt += scenario["text"]
-            from edsl.scenarios import Scenario
-
-            base_scenario = copy.copy(scenarios[0])
-            base_scenario["text"] = txt
-        return base_scenario
-
-    @staticmethod
-    def is_url(string):
-        try:
-            result = urlparse(string)
-            return all([result.scheme, result.netloc])
-        except ValueError:
-            return False
+    def from_pdf(cls, filename):
+        scenarios = list(cls.extract_text_from_pdf(filename))
+        return cls(scenarios)
 
     @classmethod
     def _from_pdf_to_image(cls, pdf_path, image_format="jpeg"):
@@ -184,7 +38,7 @@ class ScenarioListPdfMixin:
                 scenario = Scenario._from_filepath_image(image_path)
                 scenarios.append(scenario)
 
-            # print(f"Saved {len(images)} pages as images in {output_folder}")
+            print(f"Saved {len(images)} pages as images in {output_folder}")
             return cls(scenarios)
 
     @staticmethod

edsl/study/SnapShot.py

@@ -57,17 +57,10 @@ class SnapShot:
         from edsl.Base import Base
         from edsl.study.Study import Study
 
-        def is_edsl_object(obj):
-            package_name = "edsl"
-            cls = obj.__class__
-            module_name = cls.__module__
-            return module_name.startswith(package_name)
-
         for name, value in namespace.items():
             # TODO check this code logic (if there are other objects with to_dict method that are not from edsl)
             if (
-                is_edsl_object(value)
-                and hasattr(value, "to_dict")
+                hasattr(value, "to_dict")
                 and not inspect.isclass(value)
                 and value.__class__ not in [o.__class__ for o in self.exclude]
             ):