Add missing method docstring

feature/cyclic
Matthew D. Scholefield 2019-03-20 10:51:40 -05:00
parent e89677a84f
commit a8a7cea1c6
1 changed files with 2 additions and 1 deletions

View File

@ -28,7 +28,7 @@ from precise.model import create_model, ModelParams
from precise.network_runner import Listener
from precise.params import pr, save_params
from precise.train_data import TrainData
from precise.util import load_audio, glob_all, save_audio, play_audio, chunk_audio
from precise.util import load_audio, glob_all, save_audio, chunk_audio
usage = '''
Train a model on infinitely generated batches
@ -114,6 +114,7 @@ class GeneratedTrainer:
self.pos_files_it = iter(cycle(pos_files))
def layer_with(self, sample: np.ndarray, value: int) -> np.ndarray:
"""Create an identical 2d array where the second row is filled with value"""
b = np.full((2, len(sample)), value, dtype=float)
b[0] = sample
return b