"import pytorch" is the new "if-else".