Update utils/non_melting_point.py

utils/non_melting_point.py

@@ -28,16 +28,40 @@ class NgramProcessor:
             models = ["en_core_web_trf"]
         
         # Check and register curated transformer if needed
-        if "curated_transformer" not in registry.factories.get_names("pipe", "en"):
-            try:
-                # Try to import and register the curated transformer
-                import spacy_curated_transformers
-                from spacy_curated_transformers import CuratedTransformer
-                from spacy.language import Language
-                Language.factory("curated_transformer", func=CuratedTransformer)
-                tqdm.write("[NgramProcessor] Registered curated_transformer factory")
-            except ImportError:
-                tqdm.write("[NgramProcessor] Warning: spacy_curated_transformers not found. Some models may not load correctly.")
+        # Fix for registry.get_names error - check if curated_transformer exists in a compatible way
+        try:
+            # First try to check if "curated_transformer" exists in the registry
+            curated_transformer_exists = False
+            
+            # Compatible check across different spaCy versions
+            if hasattr(registry, 'factories'):
+                if hasattr(registry.factories, 'get_names'):
+                    # spaCy older version
+                    curated_transformer_exists = "curated_transformer" in registry.factories.get_names("pipe", "en")
+                elif hasattr(registry.factories, 'get'):
+                    # spaCy newer version - check differently
+                    pipe_factories = registry.factories.get("pipe", {})
+                    curated_transformer_exists = "curated_transformer" in pipe_factories
+                else:
+                    # Alternative check if structure is different
+                    factories_dict = getattr(registry.factories, "__dict__", {})
+                    for key, value in factories_dict.items():
+                        if "curated_transformer" in str(key):
+                            curated_transformer_exists = True
+                            break
+            
+            if not curated_transformer_exists:
+                try:
+                    # Try to import and register the curated transformer
+                    import spacy_curated_transformers
+                    from spacy_curated_transformers import CuratedTransformer
+                    from spacy.language import Language
+                    Language.factory("curated_transformer", func=CuratedTransformer)
+                    tqdm.write("[NgramProcessor] Registered curated_transformer factory")
+                except ImportError:
+                    tqdm.write("[NgramProcessor] Warning: spacy_curated_transformers not found. Some models may not load correctly.")
+        except Exception as e:
+            tqdm.write(f"[NgramProcessor] Warning: Error checking for curated_transformer: {str(e)}")
         
         # Load specified model
         self.models = {}
@@ -594,7 +618,6 @@ if __name__ == "__main__":
     
     # Initialize with multiple models
     processor = NgramProcessor(models=["en_core_web_trf"])
-    
     # Process with all models combined
     common_ngrams,indexed_ngrams = processor.find_filtered_ngrams(sentences)