From 242c4a0c4740d45285fdae75e2cebdb2fbb61086 Mon Sep 17 00:00:00 2001
From: Shubham Gupta <shubhamgupta3121@gmail.com>
Date: Tue, 2 Jul 2024 12:47:31 +0530
Subject: [PATCH] make usercodev4 serialized and restage protocol version

---
 packages/syft/src/syft/protocol/protocol_version.json | 5 -----
 packages/syft/src/syft/service/code/user_code.py      | 1 +
 2 files changed, 1 insertion(+), 5 deletions(-)

diff --git a/packages/syft/src/syft/protocol/protocol_version.json b/packages/syft/src/syft/protocol/protocol_version.json
index 03c68c26d54..3ca88b5c2f9 100644
--- a/packages/syft/src/syft/protocol/protocol_version.json
+++ b/packages/syft/src/syft/protocol/protocol_version.json
@@ -281,11 +281,6 @@
         }
       },
       "UserCode": {
-        "4": {
-          "version": 4,
-          "hash": "0a7181cd5f76800b6566175ffa7276d0cf38c4ddc5110114430147dfc8bfdb2a",
-          "action": "remove"
-        },
         "5": {
           "version": 5,
           "hash": "a60e08145b41f6a06822a8b6bba7419a3aaa2e8c08b6015ce072dc64aae26230",
diff --git a/packages/syft/src/syft/service/code/user_code.py b/packages/syft/src/syft/service/code/user_code.py
index 918c3468f7d..c11f9ce21b6 100644
--- a/packages/syft/src/syft/service/code/user_code.py
+++ b/packages/syft/src/syft/service/code/user_code.py
@@ -274,6 +274,7 @@ def get_sync_dependencies(self, context: AuthedServiceContext) -> list[UID]:
         return [self.user_code_link.object_uid]
 
 
+@serializable()
 class UserCodeV4(SyncableSyftObject):
     # version
     __canonical_name__ = "UserCode"