import os
import sys
from logging import getLogger
from tatau_core import settings
from tatau_core.node.worker import Worker
from tatau_core.utils.logging import configure_logging
configure_logging('worker')
logger = getLogger('tatau_core')
def load_credentials(account_address_var_name):
address = os.getenv(account_address_var_name)
if address is None:
raise ValueError('{} is not specified'.format(account_address_var_name))
storage_path = settings.KEYS_PATH
dir_name = address.replace('0x', '')
with open(os.path.join(storage_path, dir_name, 'rsa_pk.pem'), 'r') as f:
pk = f.read()
return address, pk.encode()
if __name__ == '__main__':
try:
index = '_{}'.format(sys.argv[1])
settings.TATAU_STORAGE_BASE_DIR += str(index)
except IndexError:
import os
import sys
from logging import getLogger
from tatau_core import settings
from tatau_core.contract import NodeContractInfo
from tatau_core.node import VerifierEstimator
from tatau_core.utils.logging import configure_logging
configure_logging('verifier')
logger = getLogger('tatau_core')
def load_credentials(account_address_var_name):
address = os.getenv(account_address_var_name)
if address is None:
raise ValueError(
'{} is not specified'.format(account_address_var_name))
storage_path = settings.KEYS_PATH
dir_name = address.replace('0x', '')
with open(os.path.join(storage_path, dir_name, 'rsa_pk.pem'), 'r') as f:
pk = f.read()
with open(os.path.join(storage_path, dir_name, 'wallet.json'), 'r') as f:
wallet = f.read()
with open(os.path.join(storage_path, dir_name, 'wallet.pass'), 'r') as f:
wallet_password = f.read()
import argparse
from logging import getLogger
from tatau_core.nn.benchmark.run import benchmark_train
from tatau_core.utils.logging import configure_logging
configure_logging(__name__)
logger = getLogger('tatau_core')
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Local benchmark')
parser.add_argument('-m', '--model_path', required=True, help='model path')
parser.add_argument('-train',
'--dataset_train',
required=True,
help='path to dir with train dataset chunks')
parser.add_argument('-test',
'--dataset_test',
required=True,
help='path to dir with test dataset chunks')
parser.add_argument('-e',
'--epochs',
type=int,
default=3,
help='epochs count')
parser.add_argument('-b',
'--batch_size',
type=int,
default=32,
import os
from logging import getLogger
from tatau_core import settings
from tatau_core.node.producer import Producer
from tatau_core.utils.logging import configure_logging
configure_logging('producer')
logger = getLogger('tatau_core')
def load_credentials(account_address_var_name):
address = os.getenv(account_address_var_name)
if address is None:
raise ValueError('{} is not specified'.format(account_address_var_name))
storage_path = settings.KEYS_PATH
dir_name = address.replace('0x', '')
with open(os.path.join(storage_path, dir_name, 'rsa_pk.pem'), 'r') as f:
pk = f.read()
return address, pk.encode()
def load_producer():
account_address, rsa_pk = load_credentials(
account_address_var_name='PRODUCER_ACCOUNT_ADDRESS'
)
p = Producer(