fix: remove NotImplementedError check

Author: oesteban

nitransforms/cli.py

@@ -29,10 +29,11 @@ def cli_apply(pargs):
             "Cannot determine transformation format, manually set format with the `--fmt` flag"
         )
 
-    if pargs.nonlinear:
-        xfm = nlinload(pargs.transform, fmt=fmt)
-    else:
-        xfm = linload(pargs.transform, fmt=fmt)
+    xfm = (
+        nlinload(pargs.transform, fmt=fmt)
+        if pargs.nonlinear else
+        linload(pargs.transform, fmt=fmt)
+    )
 
     # ensure a reference is set
     xfm.reference = pargs.ref or pargs.moving

nitransforms/tests/test_cli.py

@@ -57,12 +57,7 @@ def test_apply_nl(tmpdir, testdata_path):
     with pytest.raises(ValueError):
         ntcli(nlargs)
 
-    nlargs.extend(["--fmt", "afni"])
-    # no linear afni support
-    with pytest.raises(NotImplementedError):
-        ntcli(nlargs)
-
     output = "moved_from_warp.nii.gz"
-    nlargs.extend(["--nonlinear", "--out", output])
+    nlargs.extend(["--nonlinear", "--fmt", "afni", "--out", output])
     ntcli(nlargs)
     assert (tmpdir / output).check()