diff --git a/paddlehub/commands/config.py b/paddlehub/commands/config.py index da5e2e44970d3377fa253b6e85c0e88db6618a96..6045269b1d81ec225f97698cea06b09862d33fba 100644 --- a/paddlehub/commands/config.py +++ b/paddlehub/commands/config.py @@ -85,7 +85,8 @@ class ConfigCommand(BaseCommand): @staticmethod def set_log_level(level): if level not in logging._nameToLevel.keys(): - print("Allowed values include: " + str(list(logging._nameToLevel.keys()))) + print("Allowed values include: " + + str(list(logging._nameToLevel.keys()))) return with open(os.path.join(CONF_HOME, "config.json"), "r") as fp: current_config = json.load(fp) diff --git a/paddlehub/common/srv_utils.py b/paddlehub/common/srv_utils.py index f20be147f8aeb8b52f4f29e4d5e6bb39e0fe7408..9d43dc32333fd476ea5fb46499cfaeb268d11f2d 100644 --- a/paddlehub/common/srv_utils.py +++ b/paddlehub/common/srv_utils.py @@ -16,6 +16,7 @@ import os import requests import time import paddle +import socket from random import randint, seed @@ -35,10 +36,14 @@ def get_stat_server(): def hub_stat(argv): try: + s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) + s.connect(('bj.bcebos.com', 80)) + ip_addr = s.getsockname()[0] params = { 'command': ' '.join(argv), 'hub_version': version.hub_version, - 'paddle_version': paddle.__version__ + 'paddle_version': paddle.__version__, + 'ip_addr': ip_addr } stat_api = get_stat_server() r = requests.get(stat_api, params=params, timeout=0.5) diff --git a/paddlehub/module/manager.py b/paddlehub/module/manager.py index fdc714c8abde0c7e7e415ef87b053757851f2168..965b00b62c9e0007918bb3ea8939bd3c8d444dc4 100644 --- a/paddlehub/module/manager.py +++ b/paddlehub/module/manager.py @@ -116,7 +116,8 @@ class LocalModuleManager(object): print_progress=True) if module_dir: - with open(os.path.join(MODULE_HOME, module_dir, "md5.txt"), "w") as fp: + with open(os.path.join(MODULE_HOME, module_dir, "md5.txt"), + "w") as fp: fp.write(md5_value) save_path = os.path.join(MODULE_HOME, module_name) if os.path.exists(save_path):