TYP: numpyの型情報を追加 #988

sabonerune · 2024-01-07T13:39:53Z

内容

numpyのdtypeの型情報を追加してみました。

その他

かなり緩く付けたためあまり恩恵がないかもしれない。

import numpyの多くをimport numpy as npに置き換えた。

github-actions · 2024-01-07T13:43:23Z

Coverage Result

Resultを開く

Name	Stmts	Miss
run.py	514	328
voicevox_engine/init.py	1	0
voicevox_engine/cancellable_engine.py	94	72
voicevox_engine/core_adapter.py	51	17
voicevox_engine/core_initializer.py	59	30
voicevox_engine/core_wrapper.py	225	159
voicevox_engine/dev/core/init.py	2	0
voicevox_engine/dev/core/mock.py	37	8
voicevox_engine/dev/tts_engine/init.py	2	0
voicevox_engine/dev/tts_engine/mock.py	29	0
voicevox_engine/engine_manifest/EngineManifest.py	34	0
voicevox_engine/engine_manifest/EngineManifestLoader.py	12	0
voicevox_engine/engine_manifest/init.py	3	0
voicevox_engine/library_manager.py	92	5
voicevox_engine/metas/Metas.py	34	0
voicevox_engine/metas/MetasStore.py	18	6
voicevox_engine/metas/init.py	2	0
voicevox_engine/model.py	162	9
voicevox_engine/morphing.py	71	46
voicevox_engine/part_of_speech_data.py	5	0
voicevox_engine/preset/Preset.py	13	0
voicevox_engine/preset/PresetError.py	2	0
voicevox_engine/preset/PresetManager.py	80	2
voicevox_engine/preset/init.py	4	0
voicevox_engine/setting/Setting.py	11	0
voicevox_engine/setting/SettingLoader.py	17	0
voicevox_engine/setting/init.py	3	0
voicevox_engine/tts_pipeline/init.py	3	0
voicevox_engine/tts_pipeline/acoustic_feature_extractor.py	28	0
voicevox_engine/tts_pipeline/kana_converter.py	88	1
voicevox_engine/tts_pipeline/mora_list.py	7	0
voicevox_engine/tts_pipeline/text_analyzer.py	146	6
voicevox_engine/tts_pipeline/tts_engine.py	199	11
voicevox_engine/user_dict.py	145	12
voicevox_engine/utility/init.py	5	0
voicevox_engine/utility/connect_base64_waves.py	37	0
voicevox_engine/utility/core_version_utility.py	8	1
voicevox_engine/utility/mutex_utility.py	13	0
voicevox_engine/utility/path_utility.py	35	9
voicevox_engine/utility/run_utility.py	10	7
TOTAL	2301	729

sabonerune

ほぼ型情報の追加ですが挙動を変えた部分にコメントしました。

sabonerune · 2024-01-07T13:44:01Z

voicevox_engine/dev/tts_engine/mock.py

-        return wave.astype("int16")
+        return wave


エンジンの型と合わないため動作を変えました。
エンジンではnpt.NDArray[np.float32]かnpt.NDArray[np.float64]を返すはず。

ちょっと自信ないのですが、値域が違うかもです！
float32のwavは-1から1、int16は…ちょっとわかりませんが、そのあたり変換しないとすごい音量になるかもです。

そうですね。
soundfileが自動的に識別して処理していたことを忘れていました。

変更した理由
戻り値のdtypeがnp.integerやnp.int16だとmypyが通らない。

なるほどです！
たぶんfloatにした後2の15乗くらいで割ればいい感じになる気がします！（自信若干なし）

今思いついたのですがtts_engineの型情報の方を緩めればいいことに気づきました。

普通はmock側を実体に合わせるので、mockに合わせて型変えるのはちょっと逆かもです！

voicevox_engine/tts_pipeline/tts_engine.py

sabonerune · 2024-01-07T13:57:55Z

問題があったのでcloseします

Hiroshiba · 2024-01-07T14:11:40Z

closeされてますが、ぜひ型をつけたいでね！！

もし可能であれば、先に挙動が変わる部分の変更を入れた後に型追加できるとこちら的にありがたいかもです！
（そのままでもまあ大丈夫です！）

sabonerune · 2024-01-07T15:26:30Z

修正したので再度OPENします

Hiroshiba

すみません！！！
npt.NDArray -> NDArrayに、np.integerをnp.int64にする作業が可能なのか試していたのですが、間違えてpushしてしまいました 🙇 🙇 🙇

voicevox_engine/dev/tts_engine/mock.py

Hiroshiba · 2024-01-07T17:22:00Z

voicevox_engine/morphing.py

@@ -24,15 +24,15 @@
 class MorphingParameter:


この上のFIXME消せそうな気がしますね！

Hiroshiba · 2024-01-07T17:26:34Z

voicevox_engine/tts_pipeline/tts_engine.py

+    return wave * query.volumeScale
+
+
+_SoxrType = TypeVar("_SoxrType", np.float32, np.float64, np.int16, np.int32)


ここはfloatingで良くなった･･･？

Hiroshiba · 2024-01-07T17:26:49Z

voicevox_engine/tts_pipeline/tts_engine.py

@@ -203,14 +208,19 @@ def apply_output_sampling_rate(
    return wave


-def apply_output_stereo(wave: ndarray, query: AudioQuery) -> ndarray:
+T = TypeVar("T", bound=np.generic)


ここもfloatingでよくなった感じですかね

Hiroshiba

LGTM！！！

良いですね！！
ちょっとこっち側で、np.floatingや残ってるndarrayに型付けさせていただこうと思います！

今考えてるなんとなくの方針は、コアはfloat32とint64しか受け付けないのでその入出力はそれに合わせるつもりです。
あとpyorldの出力だけがfloat64になっているのですが、もうこれは別に品質に寄与しないので、全部float32で固定で良いかなと思ってます。

とにかくありがとうございました！！

…e into pr/sabonerune/988

TYP: numpyの型情報を追加

e860931

sabonerune commented Jan 7, 2024

View reviewed changes

sabonerune marked this pull request as ready for review January 7, 2024 13:49

sabonerune requested a review from a team as a code owner January 7, 2024 13:49

sabonerune requested review from y-chan and removed request for a team January 7, 2024 13:49

sabonerune closed this Jan 7, 2024

FIX: mock修正

28480d2

sabonerune reopened this Jan 7, 2024

Hiroshiba requested review from Hiroshiba and removed request for y-chan January 7, 2024 16:45

Hiroshiba added 2 commits January 8, 2024 01:52

npt.NDArray -> NDArray

8912b04

integer -> int64

53104c2

Hiroshiba reviewed Jan 7, 2024

View reviewed changes

np.floatingをいくつか排除

ea72eb8

Hiroshiba reviewed Jan 7, 2024

View reviewed changes

voicevox_engine/dev/tts_engine/mock.py Outdated Show resolved Hide resolved

Apply suggestions from code review

2b6f3ea

Hiroshiba reviewed Jan 7, 2024

View reviewed changes

voicevox_engine/morphing.py

@@ -24,15 +24,15 @@

class MorphingParameter:

Copy link

Member

Hiroshiba Jan 7, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

この上のFIXME消せそうな気がしますね！

Hiroshiba reviewed Jan 7, 2024

View reviewed changes

Hiroshiba approved these changes Jan 7, 2024

View reviewed changes

Hiroshiba added 3 commits January 8, 2024 02:30

要らなくなったはずのgenericsの撤去

5f7f349

Merge branch 'typ/numpy-type' of github.com:sabonerune/voicevox_engin…

85af577

…e into pr/sabonerune/988

lintミス

4bd562c

Hiroshiba merged commit 281f5b3 into VOICEVOX:master Jan 7, 2024
3 checks passed

Hiroshiba mentioned this pull request Jan 7, 2024

numpy.ndarrayのほぼ全てに型を付ける #989

Merged

sabonerune deleted the typ/numpy-type branch January 7, 2024 23:51

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

TYP: numpyの型情報を追加 #988

TYP: numpyの型情報を追加 #988

sabonerune commented Jan 7, 2024

github-actions bot commented Jan 7, 2024 •

edited

Loading

sabonerune left a comment

sabonerune Jan 7, 2024

Hiroshiba Jan 7, 2024

sabonerune Jan 7, 2024

sabonerune Jan 7, 2024

Hiroshiba Jan 7, 2024

sabonerune Jan 7, 2024

Hiroshiba Jan 7, 2024

sabonerune commented Jan 7, 2024

Hiroshiba commented Jan 7, 2024 •

edited

Loading

sabonerune commented Jan 7, 2024

Hiroshiba left a comment

Hiroshiba Jan 7, 2024

Hiroshiba Jan 7, 2024

Hiroshiba Jan 7, 2024

Hiroshiba left a comment

		return wave * query.volumeScale


		_SoxrType = TypeVar("_SoxrType", np.float32, np.float64, np.int16, np.int32)

TYP: numpyの型情報を追加 #988

TYP: numpyの型情報を追加 #988

Conversation

sabonerune commented Jan 7, 2024

内容

その他

github-actions bot commented Jan 7, 2024 • edited Loading

Coverage Result

sabonerune left a comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

sabonerune commented Jan 7, 2024

Hiroshiba commented Jan 7, 2024 • edited Loading

sabonerune commented Jan 7, 2024

Hiroshiba left a comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Hiroshiba left a comment

Choose a reason for hiding this comment

github-actions bot commented Jan 7, 2024 •

edited

Loading

Hiroshiba commented Jan 7, 2024 •

edited

Loading