make mypi happy

paulrobello · Jun 28, 2024 · 68d594d · 68d594d
1 parent ad8a314
commit 68d594d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/parllama/data_manager.py b/parllama/data_manager.py
@@ -221,7 +221,7 @@ def create_model(
     ) -> Iterator[Dict[str, Any]]:
         """Create a new model."""
         return ollama.create(
-            model=model_name, modelfile=model_code, quantize=quantize_level, stream=True
+            model=model_name, modelfile=str(model_code), quantize=quantize_level, stream=True
         )  # type: ignore
 
     @staticmethod

diff --git a/parllama/widgets/local_model_view.py b/parllama/widgets/local_model_view.py
@@ -1,7 +1,7 @@
 """Local Model View"""
 
 from functools import partial
-from typing import List
+from typing import List, cast
 
 from textual import on
 from textual.app import ComposeResult
@@ -169,7 +169,7 @@ def on_model_data_loaded(self, msg: LocalModelListLoaded) -> None:
         self.grid.loading = False
         if self.search_input.value:
             self.grid.filter(self.search_input.value)
-        if self.parent.has_focus:
+        if self.parent and cast(Widget, self.parent).has_focus:
             if model_name:
                 self.grid.select_by_name(model_name)
             else: