Merge remote-tracking branch 'origin' into summary_fixing

ammico/test/test_text.py

@@ -94,7 +94,7 @@ def test_text_sentiment_transformers():
     mydict["text_english"] = "I am happy that the CI is working again."
     test_obj.text_sentiment_transformers()
     assert mydict["sentiment"] == "POSITIVE"
-    assert mydict["sentiment_score"] == pytest.approx(0.99, 0.01)
+    assert mydict["sentiment_score"] == pytest.approx(0.99, 0.02)
 
 
 @pytest.mark.win_skip

ammico/test/test_utils.py

@@ -36,3 +36,22 @@ def test_dump_df(get_path):
     df = ut.dump_df(outdict)
     out_df = pd.read_csv(get_path + "example_dump_df.csv", index_col=[0])
     pd.testing.assert_frame_equal(df, out_df)
+
+
+def test_is_interactive():
+    assert ut.is_interactive
+
+
+def test_get_color_table():
+    colors = ut.get_color_table()
+    assert colors["Pink"] == {
+        "ColorName": [
+            "Pink",
+            "LightPink",
+            "HotPink",
+            "DeepPink",
+            "PaleVioletRed",
+            "MediumVioletRed",
+        ],
+        "HEX": ["#FFC0CB", "#FFB6C1", "#FF69B4", "#FF1493", "#DB7093", "#C71585"],
+    }

ammico/text.py

@@ -133,7 +133,7 @@ def text_sentiment_transformers(self):
         )
         result = pipe(self.subdict["text_english"])
         self.subdict["sentiment"] = result[0]["label"]
-        self.subdict["sentiment_score"] = result[0]["score"]
+        self.subdict["sentiment_score"] = round(result[0]["score"], 2)
 
     def text_ner(self):
         """Perform named entity recognition on the text using the Transformers pipeline."""

ammico/utils.py

@@ -2,6 +2,10 @@
 import os
 from pandas import DataFrame, read_csv
 import pooch
+import importlib_resources
+
+
+pkg = importlib_resources.files("ammico")
 
 
 class DownloadResource:
@@ -110,8 +114,9 @@ def is_interactive():
 
 
 def get_color_table():
+    path_tables = pkg / "data" / "Color_tables.csv"
     df_colors = read_csv(
-        os.path.join(os.path.dirname(__file__), "data", "Color_tables.csv"),
+        path_tables,
         delimiter=";",
         dtype=str,
         encoding="UTF-8",

pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ammico"
-version = "0.0.1"
+version = "0.1.0"
 description = "AI Media and Misinformation Content Analysis Tool"
 readme = "README.md"
 maintainers = [
@@ -31,6 +31,7 @@ dependencies = [
     "google-cloud-vision",
     "grpcio",
     "importlib_metadata",
+    "importlib_resources",
     "ipython",
     "jupyter_dash",
     "matplotlib",
@@ -71,7 +72,6 @@ documentation = "https://ssciwr.github.io/AMMICO/build/html/index.html"
 packages = ["ammico"]
 
 [tool.setuptools.package-data]
-# Include any png files found in the "data" subdirectory of "ammico"
-"ammico.data" = ["*.png"]
-mypkg = [ "*.csv"]
+# Include any png and csv files found in the "data" subdirectory of "ammico"
+"ammico.data" = ["*.png", "*.csv"]