Skip to content

Commit

Permalink
add optional type hints and None default
Browse files Browse the repository at this point in the history
  • Loading branch information
camfairchild committed Jul 25, 2022
1 parent b933f6d commit 0885919
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 17 deletions.
7 changes: 3 additions & 4 deletions bittensor/_subtensor/subtensor_impl.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,7 @@
# DEALINGS IN THE SOFTWARE.
import torch
from rich.prompt import Confirm
from typing import List, Dict, Union
from multiprocessing import Process
from typing import List, Dict, Union, Optional

import bittensor
from tqdm import tqdm
Expand Down Expand Up @@ -441,8 +440,8 @@ def register (
wait_for_finalization: bool = True,
prompt: bool = False,
max_allowed_attempts: int = 3,
num_processes: int = None,
update_interval: int = None,
num_processes: Optional[int] = None,
update_interval: Optional[int] = None,
) -> bool:
r""" Registers the wallet to chain.
Args:
Expand Down
25 changes: 12 additions & 13 deletions bittensor/utils/__init__.py
Original file line number Diff line number Diff line change
@@ -1,23 +1,22 @@
import binascii
from dataclasses import dataclass
import multiprocessing
from queue import Empty
import struct
import hashlib
from Crypto.Hash import keccak
import math
import multiprocessing
import numbers

import random
import bittensor
import ctypes
import time
import torch
import numbers
from dataclasses import dataclass
from queue import Empty
from typing import Any, Dict, Optional, Tuple, Union

import bittensor
import pandas
import requests
import torch
from Crypto.Hash import keccak
from substrateinterface import Keypair
from substrateinterface.utils import ss58
from substrateinterface import Keypair, KeypairType
from typing import Any, Tuple, List, Union, Optional, Dict


def indexed_values_to_dataframe (
Expand Down Expand Up @@ -225,7 +224,7 @@ def solve_for_nonce_block(solver: Solver, nonce_start: int, nonce_end: int, bloc
return None, time.time() - start


def solve_for_difficulty_fast( subtensor, wallet, num_processes: int = None, update_interval: int = 50_000 ) -> Optional[POWSolution]:
def solve_for_difficulty_fast( subtensor, wallet, num_processes: Optional[int] = None, update_interval: Optional[int] = None ) -> Optional[POWSolution]:
"""
Solves the POW for registration using multiprocessing.
Args:
Expand Down Expand Up @@ -379,7 +378,7 @@ def update_curr_block(block_number: int, block_bytes: bytes, diff: int, lock: mu
return solution


def create_pow( subtensor, wallet, num_processes: int = None, update_interval: int = None ) -> Optional[Dict[str, Any]]:
def create_pow( subtensor, wallet, num_processes: Optional[int] = None, update_interval: Optional[int] = None ) -> Optional[Dict[str, Any]]:
solution: POWSolution = solve_for_difficulty_fast( subtensor, wallet, num_processes=num_processes, update_interval=update_interval )
nonce, block_number, difficulty, seal = solution.nonce, solution.block_number, solution.difficulty, solution.seal
return None if nonce is None else {
Expand Down

0 comments on commit 0885919

Please sign in to comment.