cat-llm 0.0.41__tar.gz → 0.0.42__tar.gz

{cat_llm-0.0.41 → cat_llm-0.0.42}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cat-llm
-Version: 0.0.41
+Version: 0.0.42
 Summary: A tool for categorizing text data and images using LLMs and vision models
 Project-URL: Documentation, https://github.com/chrissoria/cat-llm#readme
 Project-URL: Issues, https://github.com/chrissoria/cat-llm/issues

{cat_llm-0.0.41 → cat_llm-0.0.42}/src/catllm/CERAD_functions.py

@@ -305,7 +305,7 @@ def cerad_drawn_score(
         #if no valid image path is provided
         elif  valid_image == False:
             reply = "invalid image path"
-            print("Skipped NaN input or invalid path")
+            #print("Skipped NaN input or invalid path")
             #extracted_jsons.append("""{"no_valid_path": 1}""")
             link1.append("Error processing input: {e}")
         else:
@@ -404,7 +404,8 @@ def cerad_drawn_score(
             "7": "complex_diamond",
             "8": "none"
         })
-        categorized_data['diamond_4_sides'] = np.where(categorized_data['diamond_4_sides'] > 1, 1, categorized_data['diamond_4_sides'])
+        #sometimes the model will get confused and output the number 4 instead of a 1 or 0
+        categorized_data.loc[categorized_data['diamond_4_sides'] > 1, 'diamond_4_sides'] = 1
         categorized_data['score'] = categorized_data['diamond_4_sides'] + categorized_data['diamond_equal_sides'] + categorized_data['similar'] + categorized_data['diamond_square']
 
         categorized_data.loc[categorized_data['none'] == 1, 'score'] = 0

{cat_llm-0.0.41 → cat_llm-0.0.42}/src/catllm/__about__.py

@@ -1,7 +1,7 @@
 # SPDX-FileCopyrightText: 2025-present Christopher Soria <chrissoria@berkeley.edu>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "0.0.41"
+__version__ = "0.0.42"
 __author__ = "Chris Soria"
 __email__ = "chrissoria@berkeley.edu"
 __title__ = "cat-llm"