From 0b23dcbf34fc3ba5598d98e78facc39353064141 Mon Sep 17 00:00:00 2001 From: antazoey Date: Mon, 27 Jan 2025 16:39:03 -0600 Subject: [PATCH] fix: im just dumb --- src/ape/managers/chain.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/src/ape/managers/chain.py b/src/ape/managers/chain.py index 87e4ca35d7..c0d6077f39 100644 --- a/src/ape/managers/chain.py +++ b/src/ape/managers/chain.py @@ -969,11 +969,15 @@ def get_receipt(self, transaction_hash: str) -> ReceiptAPI: def get_code(self, address: "AddressType") -> "ContractCode": network = self.provider.network - if not network.is_local: - self._code.setdefault(network.ecosystem.name, {}) - self._code[network.ecosystem.name].setdefault(network.name, {}) - if address in self._code[network.ecosystem.name][network.name]: - return self._code[network.ecosystem.name][network.name][address] + if network.is_dev: + # Avoid caching when dev, as you can manipulate the chain more + # (and there is isolation). + return self.provider.get_code(address) + + self._code.setdefault(network.ecosystem.name, {}) + self._code[network.ecosystem.name].setdefault(network.name, {}) + if address in self._code[network.ecosystem.name][network.name]: + return self._code[network.ecosystem.name][network.name][address] # Get from RPC for the first time. code = self.provider.get_code(address)