AnomalyLab 0.3.0__tar.gz → 0.3.2__tar.gz

{anomalylab-0.3.0 → anomalylab-0.3.2/AnomalyLab.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: AnomalyLab
-Version: 0.3.0
+Version: 0.3.2
 Summary: A Python package for empirical asset pricing analysis.
 Author: FinPhd
 Classifier: Programming Language :: Python :: 3

{anomalylab-0.3.0/AnomalyLab.egg-info → anomalylab-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: AnomalyLab
-Version: 0.3.0
+Version: 0.3.2
 Summary: A Python package for empirical asset pricing analysis.
 Author: FinPhd
 Classifier: Programming Language :: Python :: 3

{anomalylab-0.3.0 → anomalylab-0.3.2}/anomalylab/core/core.py

@@ -121,8 +121,7 @@ class Panel:
 
     @property
     def fm_preprocessor(self) -> FamaMacBethRegression:
-        if self._fm_preprocessor is None:
-            self._fm_preprocessor = FamaMacBethRegression(panel_data=self.panel_data)
+        self._fm_preprocessor = FamaMacBethRegression(panel_data=self.panel_data)
         return self._fm_preprocessor
 
     def format_preprocessor(self, path: str) -> FormatExcel:

{anomalylab-0.3.0 → anomalylab-0.3.2}/anomalylab/preprocess/fillna.py

@@ -201,12 +201,15 @@ class FillNa(Preprocessor):
 
     def _warning(self, fill_columns: list[str]) -> None:
         """
-        Issues warnings regarding missing values and normalization status.
+        Emit warnings regarding missing values and normalization status.
 
-        This method checks if any missing values are present in the specified
-        fill_columns. If none are found, it emits a warning. Additionally,
-        it checks if the data has already been normalized, in which case
-        it warns that missing values have been filled with zeros.
+        This method performs the following checks:
+        1. It verifies if there are any missing values in the specified `fill_columns`.
+        - If no missing values are found, a warning is issued indicating this.
+        2. It checks whether the data has already been normalized.
+        - If normalized, a warning is issued that missing values were filled with zeros during normalization.
+        3. It checks whether missing values were filled previously.
+        - If so, a warning is issued to indicate that missing values have already been handled earlier.
 
         Args:
             fill_columns (list[str]): The list of columns to check for missing values.
@@ -215,7 +218,11 @@ class FillNa(Preprocessor):
             warnings.warn(message=f"Missing values not found in {fill_columns}.")
         if self.panel_data.normalize:
             warnings.warn(
-                message=f"The data has already been normalized and the missing values have been filled 0."
+                message=f"The data has already been normalized, and missing values have been filled with 0."
+            )
+        if self.panel_data.fillna:
+            warnings.warn(
+                message=f"The missing values have already been handled earlier."
             )
 
 
@@ -224,7 +231,9 @@ if __name__ == "__main__":
 
     df: DataFrame = DataSet.get_panel_data()
 
-    panel: PanelData = PanelData(df=df, name="Stocks", classifications="industry")
+    panel: PanelData = PanelData(
+        df=df, name="Stocks", ret="return", classifications="industry"
+    )
 
     fill_nan: FillNa = FillNa(panel_data=panel)
     fill_nan.fill_group_column(

{anomalylab-0.3.0 → anomalylab-0.3.2}/anomalylab/structure/panel_data.py

@@ -182,6 +182,8 @@ class PanelData(Data):
                 raise ValueError("No classifications found.")
             check_columns = set(self.classifications)
         elif check_range == "characteristics":
+            if self.firm_characteristics is None:
+                raise ValueError("No firm characteristics found.")
             check_columns = set(self.firm_characteristics)
         else:
             raise ValueError("Invalid check_range value.")

{anomalylab-0.3.0 → anomalylab-0.3.2}/anomalylab/visualization/format.py

@@ -3,7 +3,7 @@ from anomalylab.utils.imports import *
 
 @dataclass
 class FormatExcel:
-    """Class to format Excel files by adjusting cell alignment and borders.
+    """Class to format Excel files by adjusting cell alignment, borders, and column widths.
 
     Attributes:
         path (str): The directory or file path of the Excel file(s) to format.
@@ -92,15 +92,44 @@ class FormatExcel:
                         elif direction == "to_round":
                             cell.value = cell.value.replace("[", "(").replace("]", ")")
 
+    def auto_adjust_column_widths(self):
+        """Automatically adjusts the column widths to fit the contents."""
+        for ws in self.wb.worksheets:
+            for column in ws.columns:
+                max_length = 0
+                col_letter = column[0].column_letter  # Get the column letter
+                for cell in column:
+                    try:
+                        cell_length = sum(
+                            2 if ord(char) > 127 else 1 for char in str(cell.value)
+                        )
+                        max_length = max(max_length, cell_length)
+                    except:
+                        pass
+                # Adjust for header row
+                # header_cell = ws[f"{col_letter}1"]
+                # if header_cell.value:
+                #     max_length = max(max_length, len(str(header_cell.value)))
+                adjusted_width = max_length + 2  # Add some padding to the width
+                ws.column_dimensions[col_letter].width = adjusted_width
+
     def save(self):
         """Saves the currently loaded workbook to its file path."""
         self.wb.save(self.file_path)
 
-    def process(self, align=True, line=True, convert_brackets=False):
+    def process(
+        self, align=True, line=True, convert_brackets=False, adjust_col_widths=False
+    ):
         """Processes and formats Excel files.
 
         - If the provided path is a directory, it formats all Excel files in that directory.
         - If the provided path is a file, it formats that specific Excel file.
+
+        Args:
+            align (bool): Whether to apply text alignment. Default is True.
+            line (bool): Whether to apply borders. Default is True.
+            convert_brackets (bool): Whether to convert brackets. Default is False.
+            auto_adjust (bool): Whether to adjust column widths. Default is False.
         """
         files_to_process = (
             glob(os.path.join(self.path, "*.xlsx"))
@@ -116,10 +145,12 @@ class FormatExcel:
                 self.line()
             if convert_brackets:
                 self.convert_brackets()
+            if adjust_col_widths:
+                self.auto_adjust_column_widths()
             self.save()
 
 
 if __name__ == "__main__":
     path = "..."
     excel_formatter = FormatExcel(path)
-    excel_formatter.process(convert_brackets=True)
+    excel_formatter.process()

{anomalylab-0.3.0 → anomalylab-0.3.2}/setup.py

@@ -2,7 +2,7 @@ from setuptools import find_packages, setup
 
 setup(
     name="AnomalyLab",
-    version="0.3.0",
+    version="0.3.2",
     author="FinPhd",
     # author_email="your.email@example.com",
     description="A Python package for empirical asset pricing analysis.",