from warnings import warn
def go():
print("The `dscheduler` command has been deprecated.")
print("Please use `dask-scheduler` in the future.")
print()
from distributed.cli.dask_scheduler import go
go()
if __name__ == '__main__':
go()
def go():
print("The `dscheduler` command has been deprecated.")
print("Please use `dask-scheduler` in the future.")
print()
from distributed.cli.dask_scheduler import go
go()
cmd = sys.argv[1]
sub_cmd = sys.argv[2:] if len(sys.argv) >= 3 else []
sys.argv.pop(1)
if cmd in ('help', '--help', 'h', '-h'):
help_info()
sys.exit()
sys.argv[0] = cmd # replace prog name
temp_path = os.path.join(tempfile.gettempdir(), 'rltk', 'remote')
if not os.path.exists(temp_path):
os.makedirs(temp_path, exist_ok=True)
if cmd == 'remote.worker':
logger = logging.getLogger('distributed.dask_worker')
logger.setLevel(logging.ERROR)
sys.argv.append('--local-directory')
sys.argv.append(temp_path)
# sys.argv.append('--change-directory')
sys.exit(dask_worker.go())
elif cmd == 'remote.scheduler':
logger = logging.getLogger('distributed.scheduler')
logger.setLevel(logging.ERROR)
sys.argv.append('--local-directory')
sys.argv.append(temp_path)
sys.exit(dask_scheduler.go())
else:
print('Unknown command\n')
help_info()
sys.exit()
