Cache get_decoder_class

bittensor/subtensor.py

@@ -1,6 +1,7 @@
 # The MIT License (MIT)
 # Copyright © 2021 Yuma Rao
 # Copyright © 2023 Opentensor Foundation
+import functools
 
 # Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
 # documentation files (the “Software”), to deal in the Software without restriction, including without limitation
@@ -92,6 +93,21 @@
 T = TypeVar("T")
 
 
+#######
+# Monkey patch in caching the get_decoder_class method
+#######
+if hasattr(RuntimeConfiguration, "get_decoder_class"):
+    original_get_decoder_class = RuntimeConfiguration.get_decoder_class
+
+    @functools.lru_cache(maxsize=None)
+    def cached_get_decoder_class(self, type_string):
+        return original_get_decoder_class(self, type_string)
+
+    RuntimeConfiguration.get_decoder_class = cached_get_decoder_class
+
+#######
+
+
 class ParamWithTypes(TypedDict):
     name: str  # Name of the parameter.
     type: str  # ScaleType string of the parameter.

requirements/prod.txt

@@ -26,7 +26,7 @@ pytest
 retry
 requests
 rich
-scalecodec==1.2.7
+scalecodec==1.2.7  # scalecodec should not be changed unless first verifying compatibility with the subtensor's monkeypatching of scalecodec.RuntimeConfiguration.get_decoder_class
 shtab==1.6.5
 substrate-interface==1.7.5
 termcolor

scripts/environments/apple_m1_environment.yml

@@ -229,7 +229,7 @@ dependencies:
       - retry==0.9.2
       - rich==12.5.1
       - rsa==4.9
-      - scalecodec==1.2.7
+      - scalecodec==1.2.7  # scalecodec should not be changed unless first verifying compatibility with the subtensor's monkeypatching of scalecodec.RuntimeConfiguration.get_decoder_class
       - scikit-learn==1.2.2
       - scipy==1.10.1
       - sentencepiece==0.1.99