chore: can't pickle executor we need to reinitialize it

src/concrete/ml/torch/hybrid_model.py

@@ -130,6 +130,32 @@ def __init__(
         self.optimized_linear_execution = optimized_linear_execution
         self.executor: Optional[GLWELinearLayerExecutor] = None
 
+    def __getstate__(self):
+        """Define how the instance should be pickled.
+
+        Returns:
+            dict: A dictionary containing the instance's serializable state
+        """
+        state = self.__dict__.copy()
+
+        # Remove the unpicklable executor attribute
+        if "executor" in state:
+            del state["executor"]
+
+        return state
+
+    def __setstate__(self, state):
+        """Define how the instance should be unpickled.
+
+        Args:
+            state (dict): A dictionary containing the instance's serialized state
+        """
+        self.__dict__.update(state)
+
+        # Re-initialize the executor
+        if self.optimized_linear_execution and _HAS_GLWE_BACKEND:
+            self.executor = GLWELinearLayerExecutor()
+
     def init_fhe_client(
         self, path_to_client: Optional[Path] = None, path_to_keys: Optional[Path] = None
     ):  # pragma:no cover