pozalabs / commu-code Goto Github PK

    chord_progression: list[str]
TypeError: 'type' object is not subscriptable

 commu/midi_generator/container.py         | 9 ++++-----
 commu/midi_generator/info_preprocessor.py | 4 ++--
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/commu/midi_generator/container.py b/commu/midi_generator/container.py
index 5357db4..b87d5ea 100644
--- a/commu/midi_generator/container.py
+++ b/commu/midi_generator/container.py
@@ -1,8 +1,7 @@
 import json
 from fractions import Fraction
 from pathlib import Path
-from typing import Dict, Any
-
+from typing import Dict, Any, List
 
 from pydantic import BaseModel, validator
 
@@ -21,10 +20,10 @@ class TransXlInputData(MidiMeta):
     num_generate: int
     top_k: int
     temperature: float
-    chord_progression: list[str]
+    chord_progression: List[str]
 
     @validator("chord_progression")
-    def validate_chord_progression_length(cls, value: list[str], values: Dict[str, Any]) -> list[str]:
+    def validate_chord_progression_length(cls, value: List[str], values: Dict[str, Any]) -> List[str]:
         num_measures = values.get("num_measures")
         time_signature = values.get("time_signature")
         expected_result = (num_measures - (num_measures % 4)) * Fraction(time_signature) * 8
@@ -65,4 +64,4 @@ class TransXlInputData(MidiMeta):
         return chord_token_components
 
     def to_dict(self) -> Dict[str, Any]:
-        return json.loads(self.json())
\ No newline at end of file
+        return json.loads(self.json())
diff --git a/commu/midi_generator/info_preprocessor.py b/commu/midi_generator/info_preprocessor.py
index 9368805..cd032c0 100644
--- a/commu/midi_generator/info_preprocessor.py
+++ b/commu/midi_generator/info_preprocessor.py
@@ -13,7 +13,7 @@ def encode_meta(meta_encoder: MetaEncoder, midi_meta: MidiMeta) -> List[int]:
     return meta_encoder.encode(midi_meta)
 
 
-def normalize_chord_progression(chord_progression: str) -> list[str]:
+def normalize_chord_progression(chord_progression: str) -> List[str]:
     return chord_progression.split("-")
 
 
@@ -42,4 +42,4 @@ class PreprocessTask:
             self.normalize_input_data(input_data)
 
         encoded_meta = self.preprocess()
-        return encoded_meta
\ No newline at end of file
+        return encoded_meta
-- 
2.34.1

--bpm 70 \
--audio_key aminor \
--time_signature 4/4 \
--pitch_range mid_high \
--num_measures 8 \
--inst acoustic_piano \
--genre newage \
--min_velocity 60 \
--max_velocity 80 \
--track_role main_melody \
--rhythm standard \
--chord_progression Am-Am-Am-Am-Am-Am-Am-Am-G-G-G-G-G-G-G-G-F-F-F-F-F-F-F-F-E-E-E-E-E-E-E-E-Am-Am-Am-Am-Am-Am-Am-Am-G-G-G-G-G-G-G-G-F-F-F-F-F-F-F-F-E-E-E-E-E-E-E-E \
--num_generate 3

UserWarning: Setting attributes on ParameterList is not supported.
2022-11-20 15:13:37,406 | INFO | ComMU | Generating the idx: 1
2022-11-20 15:13:39,677 | INFO | ComMU | correct_length: 8
2022-11-20 15:13:39,677 | INFO | ComMU | <list of indexes>
2022-11-20 15:13:39,677 | ERROR | ComMU | Empty sequence generated
2022-11-20 15:13:39,678 | INFO | ComMU | Generating the idx: 1
2022-11-20 15:13:41,024 | INFO | ComMU | correct_length: 8
2022-11-20 15:13:41,024 | INFO | ComMU | <list of indexes>
2022-11-20 15:13:41,025 | ERROR | ComMU | Empty sequence generated
2022-11-20 15:13:41,025 | INFO | ComMU | Generating the idx: 1
2022-11-20 15:13:42,897 | INFO | ComMU | correct_length: 8
2022-11-20 15:13:42,897 | INFO | ComMU | <list of indexes>

UserWarning: Setting attributes on ParameterList is not supported.
2022-11-20 15:30:45,402 | INFO | ComMU | Generating the idx: 1
2022-11-20 15:31:05,013 | ERROR | ComMU | remnant chord length: 8 
error in teacher forcing
2022-11-20 15:31:05,013 | INFO | ComMU | Generating the idx: 1
2022-11-20 15:31:24,684 | ERROR | ComMU | remnant chord length: 8 
error in teacher forcing