justheuristic 3 anni fa
parent
commit
5cd563995d
2 ha cambiato i file con 3 aggiunte e 5 eliminazioni
  1. 2 4
      cli/speed_test.py
  2. 1 1
      src/client/remote_sequential.py

+ 2 - 4
cli/speed_test.py

@@ -152,15 +152,13 @@ except ImportError:
     from md5 import md5
 
 try:
-    from argparse import SUPPRESS as ARG_SUPPRESS
-    from argparse import ArgumentParser as ArgParser
+    from argparse import SUPPRESS as ARG_SUPPRESS, ArgumentParser as ArgParser
 
     PARSER_TYPE_INT = int
     PARSER_TYPE_STR = str
     PARSER_TYPE_FLOAT = float
 except ImportError:
-    from optparse import SUPPRESS_HELP as ARG_SUPPRESS
-    from optparse import OptionParser as ArgParser
+    from optparse import SUPPRESS_HELP as ARG_SUPPRESS, OptionParser as ArgParser
 
     PARSER_TYPE_INT = "int"
     PARSER_TYPE_STR = "string"

+ 1 - 1
src/client/remote_sequential.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import contextlib
 import logging
 import random
-from typing import Union, Optional
+from typing import Optional, Union
 
 import torch
 from hivemind import DHT, P2P, get_logger, use_hivemind_log_handler