From b07dedd744340c2a11aefd4d40b70124df00e3d2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Mon, 15 May 2023 13:11:01 +0800
Subject: [PATCH] fix workflow (#284)

* Update extract_feature_print.py

* Update unitest.yml
---
 .github/workflows/unitest.yml | 2 +-
 extract_feature_print.py      | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/unitest.yml b/.github/workflows/unitest.yml
index ee691d1..464d799 100644
--- a/.github/workflows/unitest.yml
+++ b/.github/workflows/unitest.yml
@@ -32,5 +32,5 @@ jobs:
         touch logs/mi-test/preprocess.log
         python trainset_preprocess_pipeline_print.py logs/mute/0_gt_wavs 48000 8 logs/mi-test True
         touch logs/mi-test/extract_f0_feature.log
-        python extract_f0_print.py logs/mi-test $(nproc) pm
+        python extract_f0_print.py logs/mi-test $(nproc) pm v1
         python extract_feature_print.py cpu 1 0 0 logs/mi-test
diff --git a/extract_feature_print.py b/extract_feature_print.py
index fa46afb..abf4d25 100644
--- a/extract_feature_print.py
+++ b/extract_feature_print.py
@@ -5,11 +5,12 @@ n_part = int(sys.argv[2])
 i_part = int(sys.argv[3])
 if len(sys.argv) == 5:
     exp_dir = sys.argv[4]
+    version = sys.argv[5]
 else:
     i_gpu = sys.argv[4]
     exp_dir = sys.argv[5]
     os.environ["CUDA_VISIBLE_DEVICES"] = str(i_gpu)
-version = sys.argv[6]
+    version = sys.argv[6]
 import torch
 import torch.nn.functional as F
 import soundfile as sf