Closed oguiza closed 1 year ago
Failed test:
from tsai.basics import * from tsai.learner import all_arch_names num_classes = 5 X = torch.rand(8, 2, 50) y = torch.randint(0, num_classes, (len(X), 1, 50)) splits = TimeSplitter(show_plot=False)(y) vocab = np.arange(num_classes) fail_test = [] for arch in all_arch_names: if not "plus" in arch.lower(): continue try: fcst = TSClassifier(X, y, splits=splits, arch=arch, metrics=accuracy, vocab=vocab) with ContextManagers([fcst.no_bar(), fcst.no_logging()]): fcst.fit_one_cycle(1, 1e-3) except Exception as e: fail_test.append(arch) print(arch, e) test_eq(fail_test, [])
Closed as all all_arch_names ending in Plus pass the test.
Failed test: