Return the algorithm class with the given name.
(algorithm_name)
| 46 | |
| 47 | |
| 48 | def get_algorithm_class(algorithm_name): |
| 49 | """Return the algorithm class with the given name.""" |
| 50 | if algorithm_name not in globals(): |
| 51 | raise NotImplementedError("Algorithm not found: {}".format(algorithm_name)) |
| 52 | return globals()[algorithm_name] |
| 53 | |
| 54 | |
| 55 | class Algorithm(torch.nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected