Ver código fonte

Hotfix wrong getattr in run_server (#122)

* Hotfix wrong getattr in compression

* Remove unnecessary case of compression
Max Ryabinin 4 anos atrás
pai
commit
ee53720d38
1 arquivos alterados com 1 adições e 3 exclusões
  1. 1 3
      scripts/run_server.py

+ 1 - 3
scripts/run_server.py

@@ -59,10 +59,8 @@ def main():
     compression_name = args.pop("compression")
     if compression_name == "MEANSTD":
         compression = CompressionType.MEANSTD_LAST_AXIS_FLOAT16
-    elif compression_name == "FLOAT16":
-        compression = CompressionType.FLOAT16
     else:
-        compression = getattr(CompressionType, CompressionType.NONE)
+        compression = getattr(CompressionType, compression_name)
 
     try:
         server = Server.create(**args, optim_cls=optim_cls, start=True, verbose=True, compression=compression)