fastMONAI 0.5.0.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

fastMONAI/__init__.py

@@ -1 +1 @@
-__version__ = "0.5.0.0"
+__version__ = "0.5.1"

fastMONAI/_modidx.py

@@ -47,7 +47,20 @@ d = { 'settings': { 'branch': 'master',
                                                                                                'fastMONAI/external_data.py')},
             'fastMONAI.research_utils': { 'fastMONAI.research_utils.pred_postprocess': ( 'research_utils.html#pred_postprocess',
                                                                                          'fastMONAI/research_utils.py')},
-            'fastMONAI.utils': { 'fastMONAI.utils.ModelTrackingCallback': ('utils.html#modeltrackingcallback', 'fastMONAI/utils.py'),
+            'fastMONAI.utils': { 'fastMONAI.utils.MLflowUIManager': ('utils.html#mlflowuimanager', 'fastMONAI/utils.py'),
+                                 'fastMONAI.utils.MLflowUIManager.__init__': ('utils.html#mlflowuimanager.__init__', 'fastMONAI/utils.py'),
+                                 'fastMONAI.utils.MLflowUIManager.check_mlflow_installed': ( 'utils.html#mlflowuimanager.check_mlflow_installed',
+                                                                                             'fastMONAI/utils.py'),
+                                 'fastMONAI.utils.MLflowUIManager.find_available_port': ( 'utils.html#mlflowuimanager.find_available_port',
+                                                                                          'fastMONAI/utils.py'),
+                                 'fastMONAI.utils.MLflowUIManager.is_mlflow_running': ( 'utils.html#mlflowuimanager.is_mlflow_running',
+                                                                                        'fastMONAI/utils.py'),
+                                 'fastMONAI.utils.MLflowUIManager.is_port_available': ( 'utils.html#mlflowuimanager.is_port_available',
+                                                                                        'fastMONAI/utils.py'),
+                                 'fastMONAI.utils.MLflowUIManager.start_ui': ('utils.html#mlflowuimanager.start_ui', 'fastMONAI/utils.py'),
+                                 'fastMONAI.utils.MLflowUIManager.status': ('utils.html#mlflowuimanager.status', 'fastMONAI/utils.py'),
+                                 'fastMONAI.utils.MLflowUIManager.stop': ('utils.html#mlflowuimanager.stop', 'fastMONAI/utils.py'),
+                                 'fastMONAI.utils.ModelTrackingCallback': ('utils.html#modeltrackingcallback', 'fastMONAI/utils.py'),
                                  'fastMONAI.utils.ModelTrackingCallback.__init__': ( 'utils.html#modeltrackingcallback.__init__',
                                                                                      'fastMONAI/utils.py'),
                                  'fastMONAI.utils.ModelTrackingCallback._build_config': ( 'utils.html#modeltrackingcallback._build_config',

fastMONAI/utils.py

@@ -1,7 +1,7 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/07_utils.ipynb.
 
 # %% auto 0
-__all__ = ['store_variables', 'load_variables', 'print_colab_gpu_info', 'ModelTrackingCallback']
+__all__ = ['store_variables', 'load_variables', 'print_colab_gpu_info', 'ModelTrackingCallback', 'MLflowUIManager']
 
 # %% ../nbs/07_utils.ipynb 1
 import pickle
@@ -13,6 +13,7 @@ import os
 import tempfile
 import json
 from fastai.callback.core import Callback
+from fastcore.foundation import L
 from typing import Any
 
 # %% ../nbs/07_utils.ipynb 3
@@ -165,6 +166,8 @@ class ModelTrackingCallback(Callback):
         
         # Process each metric, handling both scalars and tensors
         for name, val in zip(metric_names, raw_metric_values):
+            if val is None:
+                continue  # Skip None values during inference
             if isinstance(val, torch.Tensor):
                 if val.numel() == 1:
                     # Single value tensor (like binary dice score)
@@ -182,8 +185,9 @@ class ModelTrackingCallback(Callback):
         
         # Handle loss values
         if len(recorder.log) >= 2:
-            metrics['train_loss'] = float(recorder.log[1])
-            if len(recorder.log) >= 3:
+            if recorder.log[1] is not None:
+                metrics['train_loss'] = float(recorder.log[1])
+            if len(recorder.log) >= 3 and recorder.log[2] is not None:
                 metrics['valid_loss'] = float(recorder.log[2])
                 
         return metrics
@@ -197,10 +201,22 @@ class ModelTrackingCallback(Callback):
         if os.path.exists(weights_file):
             mlflow.log_artifact(weights_file, "model")
         
+        # Remove MLflow callbacks before exporting learner for inference
+        # This prevents the callback from being triggered during inference
+        original_cbs = self.learn.cbs.copy()  # Save original callbacks
+        
+        # Remove ModelTrackingCallback instances from learner using proper collection type
+        filtered_cbs = L([cb for cb in self.learn.cbs if not isinstance(cb, ModelTrackingCallback)])
+        self.learn.cbs = filtered_cbs
+        
+        # Export clean learner without MLflow callbacks
         learner_path = temp_dir / "learner.pkl"
         self.learn.export(str(learner_path))
         mlflow.log_artifact(str(learner_path), "model")
         
+        # Restore original callbacks for current session
+        self.learn.cbs = original_cbs
+        
         config_path = temp_dir / "inference_settings.pkl"
         store_variables(config_path, self.size, self.reorder, self.resample)
         mlflow.log_artifact(str(config_path), "config")
@@ -235,3 +251,161 @@ class ModelTrackingCallback(Callback):
             self._register_pytorch_model()
             
         print(f"MLflow run completed. Run ID: {mlflow.active_run().info.run_id}")
+
+# %% ../nbs/07_utils.ipynb 7
+import subprocess
+import threading
+import time
+import socket
+import os
+from IPython.display import display, HTML, clear_output
+from IPython.core.magic import register_line_magic
+from IPython import get_ipython
+import requests
+import shutil
+
+class MLflowUIManager:
+    def __init__(self):
+        self.process = None
+        self.thread = None
+        self.port = 5001
+        self.host = '0.0.0.0'
+        self.backend_store_uri = './mlruns'
+        
+    def is_port_available(self, port):
+        """Check if a port is available."""
+        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+            try:
+                s.bind(('localhost', port))
+                return True
+            except OSError:
+                return False
+    
+    def is_mlflow_running(self):
+        """Check if MLflow UI is actually responding."""
+        try:
+            response = requests.get(f'http://localhost:{self.port}', timeout=2)
+            return response.status_code == 200
+        except:
+            return False
+    
+    def find_available_port(self, start_port=5001):
+        """Find an available port starting from start_port."""
+        for port in range(start_port, start_port + 10):
+            if self.is_port_available(port):
+                return port
+        return None
+    
+    def check_mlflow_installed(self):
+        """Check if MLflow is installed."""
+        return shutil.which('mlflow') is not None
+    
+    def start_ui(self, auto_open=True, quiet=False):
+        """Start MLflow UI with better error handling and user feedback."""
+        
+        # Check if MLflow is installed
+        if not self.check_mlflow_installed():
+            if not quiet:
+                display(HTML('<div style="color: #d32f2f; font-weight: bold; font-size: 14px;">❌ MLflow not installed. Run: pip install mlflow</div>'))
+            return False
+        
+        # Find available port
+        available_port = self.find_available_port(self.port)
+        if available_port is None:
+            if not quiet:
+                display(HTML('<div style="color: #d32f2f; font-weight: bold; font-size: 14px;">❌ No available ports found (5001-5010)</div>'))
+            return False
+        
+        self.port = available_port
+        
+        # Start MLflow UI in a separate thread
+        def run_mlflow():
+            try:
+                self.process = subprocess.Popen([
+                    'mlflow', 'ui', 
+                    '--host', self.host,
+                    '--port', str(self.port),
+                    '--backend-store-uri', self.backend_store_uri
+                ], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+                self.process.wait()
+            except Exception as e:
+                if not quiet:
+                    display(HTML(f'<div style="color: #d32f2f; font-weight: bold; font-size: 14px;">❌ Error: {str(e)}</div>'))
+        
+        self.thread = threading.Thread(target=run_mlflow, daemon=True)
+        self.thread.start()
+        
+        # Wait and check if server started successfully
+        max_wait = 10
+        for i in range(max_wait):
+            time.sleep(1)
+            if self.is_mlflow_running():
+                if quiet:
+                    # Bright, visible link for quiet mode
+                    display(HTML(f'''
+                        <a href="http://localhost:{self.port}" target="_blank" 
+                           style="color: #1976d2; font-weight: bold; font-size: 16px; text-decoration: underline;">
+                           🔗 MLflow UI (Port {self.port})
+                        </a>
+                    '''))
+                else:
+                    # Success message with high contrast colors
+                    display(HTML(f'''
+                        <div style="background-color: #c8e6c9; border: 2px solid #388e3c; padding: 15px; border-radius: 8px; margin: 10px 0;">
+                            <div style="color: #1b5e20; font-weight: bold; font-size: 16px; margin-bottom: 10px;">
+                                ✅ MLflow UI is running successfully!
+                            </div>
+                            <a href="http://localhost:{self.port}" target="_blank" 
+                               style="background-color: #1976d2; color: white; padding: 12px 24px; text-decoration: none; border-radius: 6px; font-weight: bold; font-size: 14px; display: inline-block; margin: 5px 0;">
+                                🔗 Open MLflow UI
+                            </a>
+                            <div style="margin-top: 10px;">
+                                <div style="color: #424242; font-size: 13px;">URL: http://localhost:{self.port}</div>
+                            </div>
+                        </div>
+                    '''))
+                return True
+        
+        # If we get here, server didn't start properly
+        if not quiet:
+            display(HTML('<div style="color: #d32f2f; font-weight: bold; font-size: 14px;">❌ Failed to start MLflow UI</div>'))
+        return False
+    
+    def stop(self):
+        """Stop the MLflow UI server."""
+        if self.process:
+            self.process.terminate()
+            self.process = None
+            display(HTML('''
+                <div style="background-color: #ffecb3; border: 2px solid #f57c00; padding: 10px; border-radius: 6px;">
+                    <span style="color: #e65100; font-weight: bold; font-size: 14px;">🛑 MLflow UI stopped</span>
+                </div>
+            '''))
+        else:
+            display(HTML('''
+                <div style="background-color: #f0f0f0; border: 2px solid #757575; padding: 10px; border-radius: 6px;">
+                    <span style="color: #424242; font-weight: bold; font-size: 14px;">ℹ️ MLflow UI is not currently running</span>
+                </div>
+            '''))
+    
+    def status(self):
+        """Check MLflow UI status."""
+        if self.is_mlflow_running():
+            display(HTML(f'''
+                <div style="background-color: #c8e6c9; border: 2px solid #388e3c; padding: 10px; border-radius: 6px;">
+                    <div style="color: #1b5e20; font-weight: bold; font-size: 14px;">✅ MLflow UI is running</div>
+                    <a href="http://localhost:{self.port}" target="_blank" 
+                       style="color: #1976d2; font-weight: bold; text-decoration: underline;">
+                       http://localhost:{self.port}
+                    </a>
+                </div>
+            '''))
+        else:
+            display(HTML('''
+                <div style="background-color: #ffcdd2; border: 2px solid #d32f2f; padding: 10px; border-radius: 6px;">
+                    <div style="color: #b71c1c; font-weight: bold; font-size: 14px;">❌ MLflow UI is not running</div>
+                    <div style="color: #424242; font-size: 13px; margin-top: 5px;">
+                        Run <code style="background-color: #f5f5f5; padding: 2px 4px; border-radius: 3px;">mlflow_ui.start_ui()</code> to start it.
+                    </div>
+                </div>
+            '''))

{fastmonai-0.5.0.0.dist-info → fastmonai-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastMONAI
-Version: 0.5.0.0
+Version: 0.5.1
 Summary: fastMONAI library
 Home-page: https://github.com/MMIV-ML/fastMONAI
 Author: Satheshkumar Kaliyugarasan

{fastmonai-0.5.0.0.dist-info → fastmonai-0.5.1.dist-info}/RECORD

@@ -1,9 +1,9 @@
-fastMONAI/__init__.py,sha256=T19ge-mjTQc59d5RqcDcri4stPYuQEaLR3cFKwPhDfI,24
-fastMONAI/_modidx.py,sha256=sVdaZs4ZogSisNkxAN1n96jaayoJqsrODsf9xodrL14,36080
+fastMONAI/__init__.py,sha256=eZ1bOun1DDVV0YLOBW4wj2FP1ajReLjbIrGmzN7ASBw,22
+fastMONAI/_modidx.py,sha256=iuYWVoaM3bmpb_Dv6lMQU8zOVPA76hYfnyFxVgKyuBg,37759
 fastMONAI/dataset_info.py,sha256=aJ-utYZ1OrA32RIQbF7jHxcDE8SgOZE3Vt1AojxnvZc,5026
 fastMONAI/external_data.py,sha256=IVj9GbIRFh9bTFkIa2wySUObSnNfZiaVtuzFxOFAi0Q,12219
 fastMONAI/research_utils.py,sha256=LZu62g8BQAVYS4dD7qDsKHJXZnDd1uLkJ6LoaMDhUhk,590
-fastMONAI/utils.py,sha256=Gr5IGb3v-tfpdFotoIaSAECPXDZS3ECOBSdvQx5vb-A,8647
+fastMONAI/utils.py,sha256=jG8SiYebcrPJsmnmMZh4SokWRj7McdJ_gftINnfcE1A,16590
 fastMONAI/vision_all.py,sha256=_l6F8ZlUaPYcplNG6mg1-1xssYforByEe4zECbPzTck,359
 fastMONAI/vision_augmentation.py,sha256=lAlrLm8jbXRmk9a6e8_o_CNTS6Pyp-KKNXwjpelUUJc,9070
 fastMONAI/vision_core.py,sha256=k4RUBzZuh9W8J4zbcVzXCKfJxkKCsBDG0oSRMwiCNp0,13848
@@ -12,9 +12,9 @@ fastMONAI/vision_inference.py,sha256=3SaJbKGbgaf9ON9PH5DtvfNlhAurov_Idnrlp4jyU9w
 fastMONAI/vision_loss.py,sha256=NrHnk1yD4EBKsp6aippppXU4l-mwmsZOqE_bsZP3ZNI,3591
 fastMONAI/vision_metrics.py,sha256=CVxdOBPaMJT6Mo5jF3WoQj6a3C-_FsnBicMAU_ZrFS8,3549
 fastMONAI/vision_plot.py,sha256=-X_nNBXx7lYCZSFBIN1587ZTA3T_-2ASBM4K31wU660,3792
-fastmonai-0.5.0.0.dist-info/licenses/LICENSE,sha256=xV8xoN4VOL0uw9X8RSs2IMuD_Ss_a9yAbtGNeBWZwnw,11337
-fastmonai-0.5.0.0.dist-info/METADATA,sha256=eO5N4RAweItcte4WhnzMkrfn4d-Vbbs19hZjqxU_TZI,7096
-fastmonai-0.5.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fastmonai-0.5.0.0.dist-info/entry_points.txt,sha256=mVBsykSXMairzzk3hJaQ8c-UiwUZqGnn4aFZ24CpsBM,40
-fastmonai-0.5.0.0.dist-info/top_level.txt,sha256=o8y7SWF9odtnIT3jvYtUn9okbJRlaAMCy7oPFCeQvQ8,10
-fastmonai-0.5.0.0.dist-info/RECORD,,
+fastmonai-0.5.1.dist-info/licenses/LICENSE,sha256=xV8xoN4VOL0uw9X8RSs2IMuD_Ss_a9yAbtGNeBWZwnw,11337
+fastmonai-0.5.1.dist-info/METADATA,sha256=VZcJOlNQR7g3mJZkMnnoukPPJZ0yt_B5T9r5yPdCLBk,7094
+fastmonai-0.5.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fastmonai-0.5.1.dist-info/entry_points.txt,sha256=mVBsykSXMairzzk3hJaQ8c-UiwUZqGnn4aFZ24CpsBM,40
+fastmonai-0.5.1.dist-info/top_level.txt,sha256=o8y7SWF9odtnIT3jvYtUn9okbJRlaAMCy7oPFCeQvQ8,10
+fastmonai-0.5.1.dist-info/RECORD,,