fix

testground/benchmark/.flake8

@@ -0,0 +1,4 @@
+[flake8]
+max-line-length = 88
+extend-ignore = E203
+exclude = .git,__pycache__,./integration_tests/contracts,./integration_tests/**/*_pb2.py

testground/benchmark/benchmark/transaction.py

@@ -55,7 +55,8 @@ def erc20_transfer_tx(nonce: int):
 
 
 Job = namedtuple(
- "Job", ["chunk", "global_seq", "num_accounts", "num_txs", "tx_type", "create_tx", "batch"]
+ "Job",
+ ["chunk", "global_seq", "num_accounts", "num_txs", "tx_type", "create_tx", "batch"],
 )
 EthTx = namedtuple("EthTx", ["tx", "raw", "sender"])
 
@@ -75,7 +76,10 @@ def _do_job(job: Job):
  print("generated", total, "txs for node", job.global_seq)
 
  # to keep it simple, only build batch inside the account
- txs = [build_cosmos_tx(*txs[start, end]) for start, end in split_batch(len(txs), job.batch)]
+ txs = [
+ build_cosmos_tx(*txs[start:end])
+ for start, end in split_batch(len(txs), job.batch)
+ ]
  acct_txs.append(txs)
  return acct_txs
 
@@ -120,15 +124,18 @@ def build_cosmos_tx(*txs: EthTx) -> str:
  """
  return base64 encoded cosmos tx, support batch
  """
- msgs = [cosmostx.build_any(
- "/ethermint.evm.v1.MsgEthereumTx",
- cosmostx.MsgEthereumTx(
- from_=tx.sender,
- raw=tx.raw,
- ),
- ) for tx in txs]
- fee = sum(tx["gas"] * tx["gasPrice"] for tx in txs)
- gas = sum(tx["gas"] for tx in txs)
+ msgs = [
+ cosmostx.build_any(
+ "/ethermint.evm.v1.MsgEthereumTx",
+ cosmostx.MsgEthereumTx(
+ from_=tx.sender,
+ raw=tx.raw,
+ ),
+ )
+ for tx in txs
+ ]
+ fee = sum(tx.tx["gas"] * tx.tx["gasPrice"] for tx in txs)
+ gas = sum(tx.tx["gas"] for tx in txs)
  body = cosmostx.TxBody(
  messages=msgs,
  extension_options=[

testground/benchmark/overlay.nix

@@ -11,6 +11,8 @@ let
  pyunormalize = [ "setuptools" ];
  pytest-github-actions-annotate-failures = [ "setuptools" ];
  cprotobuf = [ "setuptools" ];
+ flake8-black = [ "setuptools" ];
+ flake8-isort = [ "hatchling" ];
  };
  in
  lib.mapAttrs

testground/benchmark/pyproject.toml

@@ -24,6 +24,10 @@ cprotobuf = "^0.1.11"
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.2"
 pytest-github-actions-annotate-failures = "^0.2.0"
+black = "^24.10.0"
+flake8 = "^7.1.1"
+flake8-black = "^0.3.6"
+flake8-isort = "^6.1.1"
 
 [build-system]
 requires = ["poetry-core"]