feat: allow int literals with algopy.op functions that take a BigUInt…

… (currently just bsqrt)
algorandfoundation · Apr 15, 2024 · a8e14c3 · a8e14c3
1 parent 1eb3bee
commit a8e14c3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/generate_stubs.py b/scripts/generate_stubs.py
@@ -50,7 +50,7 @@ def _get_imported_name(typ: wtypes.WType) -> str:
     wtypes.bytes_wtype: bytes,
     wtypes.uint64_wtype: int,
     wtypes.account_wtype: None,  # could be str
-    wtypes.biguint_wtype: None,  # could be int
+    wtypes.biguint_wtype: int,
     wtypes.bool_wtype: None,  # already a Python type
     # below are covered transitively with respect to STACK_TYPE_MAPPING
     wtypes.application_wtype: None,  # maybe could be int? but also covered transitively anyway

diff --git a/stubs/algopy-stubs/op.pyi b/stubs/algopy-stubs/op.pyi
@@ -76,7 +76,7 @@ def bitlen(a: Bytes | UInt64 | bytes | int, /) -> UInt64:
     Native TEAL opcode: [`bitlen`](https://developer.algorand.org/docs/get-details/dapps/avm/teal/opcodes/v10/#bitlen)
     """
 
-def bsqrt(a: BigUInt, /) -> BigUInt:
+def bsqrt(a: BigUInt | int, /) -> BigUInt:
     """
     The largest integer I such that I^2 <= A. A and I are interpreted as big-endian unsigned integers