featrixsphere 0.2.4984__tar.gz → 0.2.5182__tar.gz

{featrixsphere-0.2.4984 → featrixsphere-0.2.5182}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: featrixsphere
-Version: 0.2.4984
+Version: 0.2.5182
 Summary: Transform any CSV into a production-ready ML model in minutes, not months.
 Home-page: https://github.com/Featrix/sphere
 Author: Featrix

featrixsphere-0.2.5182/VERSION

@@ -0,0 +1 @@
+0.2.5182

{featrixsphere-0.2.4984 → featrixsphere-0.2.5182}/featrixsphere/__init__.py

@@ -38,7 +38,7 @@ Example:
     ...                                labels=['Experiment A', 'Experiment B'])
 """
 
-__version__ = "0.2.4984"
+__version__ = "0.2.5182"
 __author__ = "Featrix"
 __email__ = "support@featrix.com"
 __license__ = "MIT"

{featrixsphere-0.2.4984 → featrixsphere-0.2.5182}/featrixsphere/client.py

@@ -877,24 +877,30 @@ class FeatrixSphereClient:
                     # Other HTTP errors - re-raise
                     raise
     
-    def publish_session(self, session_id: str) -> Dict[str, Any]:
+    def publish_session(self, session_id: str, org_id: str, name: str) -> Dict[str, Any]:
         """
-        Publish a session by moving it to /sphere/published/<sessionId>.
+        Publish a session by moving it to /backplane/backplane1/sphere/PUBLISHED/<org_id>/<name>/<sessionId>.
         Moves both the session file and output directory.
         
         Args:
             session_id: Session ID to publish
+            org_id: Organization ID for subdirectory organization
+            name: Name for the published session (creates subdirectory under org_id)
             
         Returns:
             Response with published_path, output_path, and status
             
         Example:
             ```python
-            result = client.publish_session("abc123")
+            result = client.publish_session("abc123", org_id="org_123", name="production_model")
             print(f"Published to: {result['published_path']}")
             ```
         """
-        response_data = self._post_json(f"/compute/session/{session_id}/publish", {})
+        data = {
+            "org_id": org_id,
+            "name": name
+        }
+        response_data = self._post_json(f"/compute/session/{session_id}/publish", data)
         return response_data
 
     def deprecate_session(self, session_id: str, warning_message: str, expiration_date: str) -> Dict[str, Any]:
@@ -1241,7 +1247,18 @@ class FeatrixSphereClient:
             } if show_live_training_movie else None
             
             while time.time() - start_time < max_wait_time:
-                session_info = self.get_session_status(session_id)
+                try:
+                    session_info = self.get_session_status(session_id)
+                except KeyboardInterrupt:
+                    print("\n\n⚠️  Interrupted by user (Ctrl+C)")
+                    print(f"   Session {session_id} status check interrupted")
+                    print("   Returning current session status...")
+                    # Return last known status or get it one more time if possible
+                    try:
+                        return self.get_session_status(session_id)
+                    except:
+                        # If we can't get status, return a basic SessionInfo
+                        return SessionInfo(session_id=session_id, status="unknown", jobs={})
                 
                 # Clear previous output and show updated status
                 clear_output(wait=True)
@@ -1357,7 +1374,19 @@ class FeatrixSphereClient:
                 session_task = progress.add_task(f"[bold green]Session {session_id}", total=100)
                 
                 while time.time() - start_time < max_wait_time:
-                    session_info = self.get_session_status(session_id)
+                    try:
+                        session_info = self.get_session_status(session_id)
+                    except KeyboardInterrupt:
+                        progress.console.print("\n\n[bold yellow]⚠️  Interrupted by user (Ctrl+C)[/bold yellow]")
+                        progress.console.print(f"   Session {session_id} status check interrupted")
+                        progress.console.print("   Returning current session status...")
+                        # Return last known status or get it one more time if possible
+                        try:
+                            return self.get_session_status(session_id)
+                        except:
+                            # If we can't get status, return a basic SessionInfo
+                            from featrixsphere.client import SessionInfo
+                            return SessionInfo(session_id=session_id, status="unknown", jobs={})
                     
                     # Update session progress
                     elapsed = time.time() - start_time
@@ -1426,7 +1455,19 @@ class FeatrixSphereClient:
         jobs_appeared = False
         
         while time.time() - initial_wait_start < initial_wait_timeout:
-            session_info = self.get_session_status(session_id)
+            try:
+                session_info = self.get_session_status(session_id)
+            except KeyboardInterrupt:
+                logger.info("\n⚠️  Interrupted by user (Ctrl+C)")
+                logger.info(f"   Session {session_id} status check interrupted")
+                logger.info("   Returning current session status...")
+                # Return last known status or get it one more time if possible
+                try:
+                    return self.get_session_status(session_id)
+                except:
+                    # If we can't get status, return a basic SessionInfo
+                    return SessionInfo(session_id=session_id, status="unknown", jobs={})
+            
             if session_info.jobs:
                 jobs_appeared = True
                 break
@@ -1434,7 +1475,20 @@ class FeatrixSphereClient:
         
         # Main monitoring loop
         while time.time() - start_time < max_wait_time:
-            session_info = self.get_session_status(session_id)
+            try:
+                session_info = self.get_session_status(session_id)
+            except KeyboardInterrupt:
+                logger.info("\n⚠️  Interrupted by user (Ctrl+C)")
+                logger.info(f"   Session {session_id} status check interrupted")
+                logger.info("   Returning current session status...")
+                # Return last known status or get it one more time if possible
+                try:
+                    return self.get_session_status(session_id)
+                except:
+                    # If we can't get status, return a basic SessionInfo
+                    from featrixsphere.client import SessionInfo
+                    return SessionInfo(session_id=session_id, status="unknown", jobs={})
+            
             elapsed = time.time() - start_time
             
             # Call the callback with current status
@@ -1471,7 +1525,19 @@ class FeatrixSphereClient:
         last_num_lines = 0
         
         while time.time() - start_time < max_wait_time:
-            session_info = self.get_session_status(session_id)
+            try:
+                session_info = self.get_session_status(session_id)
+            except KeyboardInterrupt:
+                print("\n\n⚠️  Interrupted by user (Ctrl+C)")
+                print(f"   Session {session_id} status check interrupted")
+                print("   Returning current session status...")
+                # Return last known status or get it one more time if possible
+                try:
+                    return self.get_session_status(session_id)
+                except:
+                    # If we can't get status, return a basic SessionInfo
+                    from featrixsphere.client import SessionInfo
+                    return SessionInfo(session_id=session_id, status="unknown", jobs={})
             
             # Clear previous lines if terminal supports it
             if sys.stdout.isatty() and last_num_lines > 0:
@@ -2866,6 +2932,110 @@ class FeatrixSphereClient:
         response_data = self._post_json(f"/session/{session_id}/predict", request_payload, max_retries=max_retries)
         return response_data
     
+    def explain(self, session_id: str, record, class_idx: int = None,
+                target_column: str = None, predictor_id: str = None,
+                record_b: Dict[str, Any] = None,
+                max_retries: int = None) -> Dict[str, Any]:
+        """
+        Explain a prediction using gradient attribution.
+        
+        Supports multiple modes:
+        - explain(record): Explain a single row
+        - explain(record, record_b=other_record): Compare two rows
+        - explain([record1, record2, ...]): Explain multiple rows
+        
+        Returns what matters to Featrix in the given row(s):
+        - Which features mattered for this prediction
+        - Which relationships mattered for this prediction
+        
+        Args:
+            session_id: ID of session with trained predictor
+            record: Record dictionary (without target column), or list of records
+            class_idx: Target class index for attribution (default: predicted class)
+            target_column: Specific target column predictor to use (optional)
+            predictor_id: Specific predictor ID to use (optional)
+            record_b: Optional second record for comparison
+            max_retries: Number of retries for errors (default: uses client default)
+            
+        Returns:
+            For single record:
+                Dictionary with:
+                    - feature_scores: {col_name: score} - gradient norm per feature
+                    - pair_scores: {(i, j): score} - gradient norm per relationship pair
+                    - target_class_idx: The class index used for attribution
+                    - logit: The prediction logit
+            
+            For two records (record_b provided):
+                Dictionary with:
+                    - record_a: Explanation for first record
+                    - record_b: Explanation for second record
+                    - difference: Difference in feature_scores and pair_scores
+            
+            For list of records:
+                Dictionary with:
+                    - explanations: List of explanation dictionaries, one per record
+        """
+        # Clean NaN/Inf values
+        if isinstance(record, list):
+            cleaned_record = [self.replace_nans_with_nulls(self._clean_numpy_values(r)) for r in record]
+        else:
+            cleaned_record = self.replace_nans_with_nulls(self._clean_numpy_values(record))
+        
+        cleaned_record_b = None
+        if record_b is not None:
+            cleaned_record_b = self.replace_nans_with_nulls(self._clean_numpy_values(record_b))
+        
+        # Build request payload
+        request_payload = {
+            "query_record": cleaned_record,
+        }
+        
+        if class_idx is not None:
+            request_payload["class_idx"] = class_idx
+        if target_column:
+            request_payload["target_column"] = target_column
+        if predictor_id:
+            request_payload["predictor_id"] = predictor_id
+        if cleaned_record_b is not None:
+            request_payload["query_record_b"] = cleaned_record_b
+        
+        response_data = self._post_json(f"/session/{session_id}/explain", request_payload, max_retries=max_retries)
+        
+        # Helper to convert pair_scores keys back to tuples
+        def convert_pair_scores(ps_dict):
+            if not isinstance(ps_dict, dict):
+                return ps_dict
+            result = {}
+            for key, score in ps_dict.items():
+                # Key format is "i_j"
+                parts = key.split("_")
+                if len(parts) == 2:
+                    try:
+                        i, j = int(parts[0]), int(parts[1])
+                        result[(i, j)] = score
+                    except ValueError:
+                        result[key] = score
+                else:
+                    result[key] = score
+            return result
+        
+        # Convert pair_scores keys back to tuples for easier use
+        if "pair_scores" in response_data:
+            response_data["pair_scores"] = convert_pair_scores(response_data["pair_scores"])
+        elif "explanations" in response_data:
+            for expl in response_data["explanations"]:
+                if "pair_scores" in expl:
+                    expl["pair_scores"] = convert_pair_scores(expl["pair_scores"])
+        elif "record_a" in response_data:
+            if "pair_scores" in response_data["record_a"]:
+                response_data["record_a"]["pair_scores"] = convert_pair_scores(response_data["record_a"]["pair_scores"])
+            if "pair_scores" in response_data["record_b"]:
+                response_data["record_b"]["pair_scores"] = convert_pair_scores(response_data["record_b"]["pair_scores"])
+            if "difference" in response_data and "pair_scores" in response_data["difference"]:
+                response_data["difference"]["pair_scores"] = convert_pair_scores(response_data["difference"]["pair_scores"])
+        
+        return response_data
+    
     def get_training_metrics(self, session_id: str, max_retries: int = None) -> Dict[str, Any]:
         """
         Get training metrics for a session's single predictor.
@@ -3985,108 +4155,6 @@ class FeatrixSphereClient:
         
         return predictor_id
     
-    def _resolve_predictor_id(self, session_id: str, predictor_id: str = None, target_column: str = None, debug: bool = False) -> Dict[str, Any]:
-        """
-        Resolve predictor_id or target_column to predictor information.
-        
-        Args:
-            session_id: Session ID to check
-            predictor_id: Specific predictor ID to resolve
-            target_column: Target column name (fallback if predictor_id not provided)
-            debug: Whether to print debug information
-            
-        Returns:
-            Dictionary with predictor info including target_column, path, predictor_id
-            
-        Raises:
-            ValueError: If predictor not found or ambiguous
-        """
-        available_predictors = self._get_available_predictors(session_id, debug=debug)
-        
-        if not available_predictors:
-            # Don't fail here - let the server try to find/auto-discover the predictor
-            # The server's /predict endpoint has smart fallback logic to find checkpoint files
-            # even if the session file wasn't properly updated (e.g., training crashed)
-            if debug:
-                print(f"⚠️  No predictors found via models endpoint, letting server handle discovery")
-            return {
-                'target_column': target_column,
-                'predictor_id': predictor_id,
-                'path': None,
-                'type': None,
-                'server_discovery': True  # Flag that server should auto-discover
-            }
-        
-        # If predictor_id is provided, find it directly (since it's now the key)
-        if predictor_id:
-            if predictor_id in available_predictors:
-                predictor_info = available_predictors[predictor_id]
-                return {
-                    'target_column': predictor_info.get('target_column'),
-                    'predictor_id': predictor_id,
-                    'path': predictor_info.get('path'),
-                    'type': predictor_info.get('type')
-                }
-            
-            # Predictor ID not found
-            all_predictor_ids = list(available_predictors.keys())
-            
-            raise ValueError(
-                f"Predictor ID '{predictor_id}' not found for session {session_id}. "
-                f"Available predictor IDs: {all_predictor_ids}"
-            )
-        
-        # Fallback to target_column validation (search through values)
-        if target_column is None:
-            # Auto-detect: only valid if there's exactly one predictor
-            if len(available_predictors) == 1:
-                predictor_id = list(available_predictors.keys())[0]
-                predictor_info = available_predictors[predictor_id]
-                return {
-                    'target_column': predictor_info.get('target_column'),
-                    'predictor_id': predictor_id,
-                    'path': predictor_info.get('path'),
-                    'type': predictor_info.get('type')
-                }
-            else:
-                # Show unique target columns for clarity
-                target_columns = list(set(pred.get('target_column') for pred in available_predictors.values()))
-                raise ValueError(
-                    f"Multiple predictors found for session {session_id} with target columns: {target_columns}. "
-                    f"Please specify predictor_id parameter for precise selection."
-                )
-        else:
-            # Find predictors by target column (there might be multiple)
-            matching_predictors = {
-                pred_id: pred_info for pred_id, pred_info in available_predictors.items()
-                if pred_info.get('target_column') == target_column
-            }
-            
-            if not matching_predictors:
-                target_columns = list(set(pred.get('target_column') for pred in available_predictors.values()))
-                raise ValueError(
-                    f"No trained predictor found for target column '{target_column}' in session {session_id}. "
-                    f"Available target columns: {target_columns}"
-                )
-            
-            if len(matching_predictors) == 1:
-                # Only one predictor for this target column
-                predictor_id = list(matching_predictors.keys())[0]
-                predictor_info = matching_predictors[predictor_id]
-                return {
-                    'target_column': target_column,
-                    'predictor_id': predictor_id,
-                    'path': predictor_info.get('path'),
-                    'type': predictor_info.get('type')
-                }
-            else:
-                # Multiple predictors for the same target column
-                predictor_ids = list(matching_predictors.keys())
-                raise ValueError(
-                    f"Multiple predictors found for target column '{target_column}' in session {session_id}: {predictor_ids}. "
-                    f"Please specify predictor_id parameter for precise selection."
-                )
-    
     def list_predictors(self, session_id: str, verbose: bool = True, debug: bool = False) -> Dict[str, Dict[str, Any]]:
         """
         List all available predictors in a session and their target columns.

{featrixsphere-0.2.4984 → featrixsphere-0.2.5182}/featrixsphere.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: featrixsphere
-Version: 0.2.4984
+Version: 0.2.5182
 Summary: Transform any CSV into a production-ready ML model in minutes, not months.
 Home-page: https://github.com/Featrix/sphere
 Author: Featrix

{featrixsphere-0.2.4984 → featrixsphere-0.2.5182}/featrixsphere.egg-info/SOURCES.txt

@@ -8,7 +8,6 @@ setup.py
 system_monitor.py
 featrixsphere/__init__.py
 featrixsphere/client.py
-featrixsphere/test_client.py
 featrixsphere.egg-info/PKG-INFO
 featrixsphere.egg-info/SOURCES.txt
 featrixsphere.egg-info/dependency_links.txt
@@ -63,13 +62,6 @@ src/start_celery_gpu_worker.sh
 src/start_celery_movie_worker.sh
 src/tail-watch.py
 src/test_api_client.py
-src/test_complete_workflow.py
-src/test_json_tables_prediction.py
-src/test_redis_predictions.py
-src/test_server_connection.py
-src/test_session_models.py
-src/test_single_predictor_api.py
-src/test_upload_endpoint.py
 src/tree.py
 src/utils.py
 src/version.py
@@ -109,8 +101,11 @@ src/lib/weightwatcher_tracking.py
 src/lib/featrix/__init__.py
 src/lib/featrix/neural/__init__.py
 src/lib/featrix/neural/calibration_utils.py
+src/lib/featrix/neural/causal_relationship_scorer.py
+src/lib/featrix/neural/charting.py
 src/lib/featrix/neural/classification_metrics.py
 src/lib/featrix/neural/config.py
+src/lib/featrix/neural/customer_quality_tracker.py
 src/lib/featrix/neural/data_frame_data_set.py
 src/lib/featrix/neural/dataloader_utils.py
 src/lib/featrix/neural/detect.py
@@ -148,6 +143,8 @@ src/lib/featrix/neural/json_cache.py
 src/lib/featrix/neural/json_codec.py
 src/lib/featrix/neural/local_string_cache.py
 src/lib/featrix/neural/logging_config.py
+src/lib/featrix/neural/lr_timeline.py
+src/lib/featrix/neural/mask_bias_tracker.py
 src/lib/featrix/neural/mask_tracker.py
 src/lib/featrix/neural/model_config.py
 src/lib/featrix/neural/model_hash.py
@@ -160,6 +157,7 @@ src/lib/featrix/neural/prng_control.py
 src/lib/featrix/neural/relationship_complexity.py
 src/lib/featrix/neural/relationship_estimator.py
 src/lib/featrix/neural/relationship_extractor.py
+src/lib/featrix/neural/relationship_importance_validator.py
 src/lib/featrix/neural/relationship_performance.py
 src/lib/featrix/neural/relationship_preanalysis.py
 src/lib/featrix/neural/relationship_search.py
@@ -178,7 +176,6 @@ src/lib/featrix/neural/string_analysis.py
 src/lib/featrix/neural/string_cache.py
 src/lib/featrix/neural/string_codec.py
 src/lib/featrix/neural/string_list_codec.py
-src/lib/featrix/neural/test_graph_encoder.py
 src/lib/featrix/neural/timestamp_codec.py
 src/lib/featrix/neural/training_banner.py
 src/lib/featrix/neural/training_context_manager.py
@@ -213,6 +210,9 @@ src/lib/featrix/neural/qa/test_focal_comparison_enhanced.py
 src/lib/featrix/neural/qa/test_focal_loss_single_predictor.py
 src/lib/featrix/neural/qa/test_hybrid_columns.py
 src/lib/featrix/neural/qa/test_label_smoothing.py
+src/lib/featrix/neural/qa/test_lift_measurement_validation.py
+src/lib/featrix/neural/qa/test_lr_timeline_smoothness.py
+src/lib/featrix/neural/qa/test_min_support_rank_scaling.py
 src/lib/featrix/neural/qa/test_monitor_integration.py
 src/lib/featrix/neural/qa/test_piecewise_epochs.py
 src/lib/featrix/neural/qa/test_predict_during_training.py
@@ -222,9 +222,4 @@ src/lib/featrix/neural/qa/test_set_encoder_strategies.py
 src/lib/featrix/neural/qa/test_timeline_quick.py
 src/lib/featrix/neural/qa/test_training_monitor.py
 src/lib/featrix/neural/qa/test_warning_tracking.py
-src/lib/featrix/neural/qa/visualize_training_timeline.py
-tests/test_client_data.py
-tests/test_client_predictions.py
-tests/test_client_sessions.py
-tests/test_client_training.py
-tests/test_local_integration.py
+src/lib/featrix/neural/qa/visualize_training_timeline.py