added baml test fix and format

.github/workflows/basic_tests.yml

@@ -189,6 +189,9 @@ jobs:
         with:
           python-version: ${{ inputs.python-version }}
 
+      - name: Install BAML Dependencies
+        run: uv add "cognee[baml]"
+
       - name: Run Simple Examples
         run: uv run python ./examples/python/simple_example.py
 

cognee/infrastructure/data/utils/extract_keywords.py

@@ -1,4 +1,3 @@
-
 from cognee.infrastructure.data.exceptions.exceptions import KeywordExtractionError
 
 

cognee/shared/utils.py

@@ -20,8 +20,10 @@ proxy_url = "https://test.prometh.ai"
 def get_entities(tagged_tokens):
     try:
         import nltk
+
         nltk.download("maxent_ne_chunker", quiet=True)
         from nltk.chunk import ne_chunk
+
         return ne_chunk(tagged_tokens)
     except ImportError:
         raise ImportError(