Coverage for src/nos/benchmarks/benchmark.py: 100%

1from dataclasses import ( 

2 dataclass, 

3 field, 

4) 

5from typing import ( 

6 List, 

7) 

8 

9from continuiti.data import ( 

10 OperatorDataset, 

11) 

12 

13from nos.metrics import ( 

14 L1Error, 

15 Metric, 

16 MSError, 

17 NumberOfParameters, 

18 SpeedOfEvaluation, 

19) 

20 

21 

22@dataclass 

23class Benchmark: 

24 train_set: OperatorDataset 

25 test_set: OperatorDataset 

26 metrics: List[Metric] = field( 

27 default_factory=lambda: [L1Error(), MSError(), NumberOfParameters(), SpeedOfEvaluation()] 

28 )