update

local_test.py

@@ -0,0 +1,64 @@
+import os
+import tempfile
+import shutil
+
+if os.path.exists("/tmp/submit_private"):
+  shutil.rmtree("/tmp/submit_private")
+
+with tempfile.TemporaryDirectory() as tempdir:
+  config_path = os.path.join(tempdir, "config.json")
+  
+  assert not os.system(f"ssh-keygen -f {tempdir}/ssh-key-ecdsa -t ecdsa -b 521 -q -N \"\"")
+
+  config = """
+  model: whisper
+  model_key: whisper
+  config.json:
+      name: 'faster-whisper-server:latest'
+      support_devices:
+      - cpu
+      model_path: ''
+      port: 8080
+      other_ports: []
+      other_ports_count: 1
+      entrypoint: start.bat
+      MIN_CHUNK: 2.5
+      MIN_ADD_CHUNK: 2.5
+      COMPUTE_TYPE: int8
+      NUM_WORKERS: 1
+      CPU_THREADS: 2
+      BEAM_SIZE: 5
+      BATCH: 1
+      LANG: auto
+      DEVICE: cpu
+      CHUNK_LENGTH: 5
+      CLASS_MODEL: ./models/faster-whisper-base
+      EN_MODEL: ./models/faster-whisper-base
+      ZH_MODEL: ./models/faster-whisper-base
+      RU_MODEL: ./models/faster-whisper-base
+      PT_MODEL: ./models/faster-whisper-base
+      AR_MODEL: ./models/faster-whisper-base
+      NEW_VERSION: 1
+      NEED_RESET: 0
+  leaderboard_options:
+    nfs:
+      - name: whisper
+        srcRelativePath: leaderboard/pc_asr/en.tar.gz
+        mountPoint: /tmp
+        source: ceph_customer
+  """
+
+  with open(config_path, "w") as f:
+    f.write(config)
+
+  os.environ["SSH_KEY_DIR"] = tempdir
+  os.environ["SUBMIT_CONFIG_FILEPATH"] = config_path
+  os.environ["MODEL_MAPPING"] = '{"whisper": "edge-ml.tar.gz"}'
+
+  from run_async_a10 import get_sut_url_windows
+
+
+  print(get_sut_url_windows())
+  
+  import time
+  time.sleep(3600)