featrixsphere 0.2.5563__tar.gz → 0.2.5566__tar.gz

{featrixsphere-0.2.5563 → featrixsphere-0.2.5566}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: featrixsphere
-Version: 0.2.5563
+Version: 0.2.5566
 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.5566/VERSION

@@ -0,0 +1 @@
+0.2.5566

{featrixsphere-0.2.5563 → featrixsphere-0.2.5566}/featrixsphere/__init__.py

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

{featrixsphere-0.2.5563 → featrixsphere-0.2.5566}/featrixsphere/client.py

@@ -215,18 +215,18 @@ class PredictionBatch:
         )
         
         # Populate cache with results
-        predictions = batch_results.get('results', {})
+        predictions = batch_results.get('predictions', [])
         successful = 0
         failed = 0
-        
-        for queue_id, prediction in predictions.items():
+
+        for prediction in predictions:
             if isinstance(prediction, dict):
                 row_index = prediction.get('row_index', 0)
                 if row_index < len(records):
                     record = records[row_index]
                     record_hash = self._hash_record(record)
                     self._cache[record_hash] = prediction
-                
+
                 if prediction.get('prediction') is not None:
                     successful += 1
                 else:

{featrixsphere-0.2.5563 → featrixsphere-0.2.5566}/featrixsphere.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: featrixsphere
-Version: 0.2.5563
+Version: 0.2.5566
 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.5563 → featrixsphere-0.2.5566}/src/lib/featrix/neural/embedded_space.py

@@ -6852,6 +6852,8 @@ class EmbeddingSpace(object):
                     # Also save state_dict for backward compatibility
                     state['encoder_state_dict'] = value.state_dict()
                     logger.info(f"💾 Saved encoder as full object (preserves exact architecture) + state_dict backup")
+                    logger.info(f"   DEBUG: state['encoder'] type = {type(state['encoder']).__name__}, is None = {state['encoder'] is None}")
+                    logger.info(f"   DEBUG: state['encoder_state_dict'] has {len(state['encoder_state_dict'])} keys")
                     continue
                 except Exception as e:
                     logger.warning(f"Failed to save encoder as object, falling back to state_dict only: {e}")
@@ -6962,7 +6964,12 @@ class EmbeddingSpace(object):
         for key in keys_to_remove:
             logger.debug(f"Removing {key} from pickle state (contains sqlite3.Connection)")
             del state[key]
-        
+
+        # Final debug: log what keys we're returning
+        logger.info(f"💾 __getstate__ returning state with keys: {list(state.keys())}")
+        logger.info(f"   'encoder' in state: {'encoder' in state}")
+        logger.info(f"   'encoder_state_dict' in state: {'encoder_state_dict' in state}")
+
         return state
     
     def __setstate__(self, state):
@@ -7282,32 +7289,19 @@ class EmbeddingSpace(object):
 
                 # ================================================================
                 # PREDICTOR ATTENTION DETECTION
-                # Check if checkpoint was trained WITH or WITHOUT predictor attention.
-                # If checkpoint lacks attention layers but SphereConfig has attention enabled,
-                # we must temporarily disable it to avoid 511 missing keys error.
+                # Detect from checkpoint whether attention was used, then set it
+                # explicitly on the config so SimpleMLP doesn't read from SphereConfig.
                 # ================================================================
                 checkpoint_has_attention = any('.attention_layers.' in key for key in encoder_state_dict.keys())
-                sphere_config = SphereConfig.get_instance()
-                current_attention_setting = sphere_config.get_enable_predictor_attention()
-
-                if current_attention_setting and not checkpoint_has_attention:
-                    logger.warning(f"   🔧 PREDICTOR ATTENTION FIX: Checkpoint was trained WITHOUT attention layers")
-                    logger.warning(f"      Current SphereConfig has enable_predictor_attention=True")
-                    logger.warning(f"      Temporarily disabling predictor attention to match checkpoint architecture")
-                    # Temporarily override SphereConfig to disable attention during encoder creation
-                    # This is safe because SimpleMLP.__init__ reads from SphereConfig when config.use_attention is None
-                    sphere_config._config['enable_predictor_attention'] = False
-                    self._attention_was_disabled_for_loading = True  # Track so we can restore later if needed
-                elif checkpoint_has_attention and not current_attention_setting:
-                    logger.warning(f"   🔧 PREDICTOR ATTENTION FIX: Checkpoint was trained WITH attention layers")
-                    logger.warning(f"      Current SphereConfig has enable_predictor_attention=False")
-                    logger.warning(f"      Temporarily enabling predictor attention to match checkpoint architecture")
-                    sphere_config._config['enable_predictor_attention'] = True
-                    self._attention_was_disabled_for_loading = False  # Actually enabled
-                elif checkpoint_has_attention:
-                    logger.info(f"   ✅ Predictor attention: checkpoint and config both have attention ENABLED")
-                else:
-                    logger.info(f"   ✅ Predictor attention: checkpoint and config both have attention DISABLED")
+                logger.info(f"   🔍 Checkpoint attention detection: {'ENABLED' if checkpoint_has_attention else 'DISABLED'}")
+
+                # Set use_attention explicitly on the config objects to match checkpoint
+                # This prevents SimpleMLP from reading SphereConfig (which may differ)
+                if hasattr(self.encoder_config, 'joint_predictor_config'):
+                    self.encoder_config.joint_predictor_config.use_attention = checkpoint_has_attention
+                    logger.info(f"   ✅ Set joint_predictor_config.use_attention = {checkpoint_has_attention}")
+                if hasattr(self.encoder_config, 'joint_predictor_short_config'):
+                    self.encoder_config.joint_predictor_short_config.use_attention = checkpoint_has_attention
 
                 # Recreate the encoder (same as in __init__)
                 logger.info(f"   🔧 Creating FeatrixTableEncoder from {len(self.col_codecs)} codecs...")

featrixsphere-0.2.5563/VERSION

@@ -1 +0,0 @@
-0.2.5563