Update lightest.py

* Added corrections
jzsmoreno · Oct 3, 2023 · a6473ba · a6473ba
1 parent 0c0624f
commit a6473ba
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/pydbsmgr/lightest.py b/pydbsmgr/lightest.py
@@ -21,9 +21,20 @@ def clean_frame(self) -> DataFrame:
                 else:
                     table[cols[column_index]] = table[cols[column_index]].apply(clean)
                     table[cols[column_index]] = table[cols[column_index]].apply(remove_char)
-                    table[cols[column_index]] = table[cols[column_index]].apply(
-                        lambda text: text.title()
-                    )
+                    try:
+                        table[cols[column_index]] = table[cols[column_index]].apply(
+                            lambda text: text.title()
+                        )
+                    except AttributeError as e:
+                        warning_type = "UserWarning"
+                        msg = (
+                            "It was not possible to perform the cleaning, the column {%s} is duplicated. "
+                            % cols[column_index]
+                        )
+                        msg += "Error: {%s}" % e
+                        print(f"{warning_type}: {msg}")
+                        sys.exit("Perform correction manually")
+
                     table[cols[column_index]] = table[cols[column_index]].apply(
                         self._correct_str, datatype=datatype
                     )

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="pydbsmgr",
-    version="0.5.0",
+    version="0.5.1",
     author="J. A. Moreno-Guerra",
     author_email="[email protected]",
     description="Testing installation of Package",
@@ -30,6 +30,7 @@
         "pyarrow",
         "fastparquet",
         "loguru",
+        "psutil",
     ],
     classifiers=[
         "Programming Language :: Python :: 3",