Upgrade tapir Coordinator

nucypher · Sep 4, 2024 · 2b9f812 · 2b9f812
1 parent b294524
commit 2b9f812
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/deployment/artifacts/tapir.json b/deployment/artifacts/tapir.json
@@ -4350,6 +4350,13 @@
                     ],
                     "outputs": []
                 },
+                {
+                    "type": "function",
+                    "name": "initializeNumberOfRituals",
+                    "stateMutability": "nonpayable",
+                    "inputs": [],
+                    "outputs": []
+                },
                 {
                     "type": "function",
                     "name": "initiateRitual",
@@ -4655,7 +4662,7 @@
                     "stateMutability": "view",
                     "inputs": [
                         {
-                            "name": "",
+                            "name": "ritualId",
                             "type": "uint256",
                             "internalType": "uint256"
                         }

diff --git a/scripts/tapir/upgrade_coordinator.py b/scripts/tapir/upgrade_coordinator.py
@@ -19,8 +19,13 @@ def main():
     deployer = Deployer.from_yaml(filepath=CONSTRUCTOR_PARAMS_FILEPATH, verify=VERIFY)
     instances = contracts_from_registry(filepath=ARTIFACTS_DIR / "tapir.json", chain_id=80002)
 
-    coordinator = deployer.upgrade(
-        project.Coordinator, instances[project.Coordinator.contract_type.name].address
+    implementation = deployer.deploy(project.Coordinator)
+    encoded_initializer_function = implementation.initializeNumberOfRituals.encode_input()
+    encoded_initializer_function = b""
+    coordinator = deployer.upgradeTo(
+        implementation,
+        instances[project.Coordinator.contract_type.name].address,
+        encoded_initializer_function,
     )
 
     deployments = [