              )
          )
  
          was_trained = run_test(
              "Train lightweight model for 5 iterations WITHOUT WTL.",
              train_args(
                  "lightweight", pathjoin(vid_base, "model"),
                  pathjoin(vid_base, "faces"), pathjoin(vid_base, "test_alignments.fsa")
              )
          )
 
</s> remove                 "lightweight", pathjoin(vid_base, "model"),
                pathjoin(vid_base, "faces"), pathjoin(vid_base, "test_alignments.fsa")
 </s> add                 "lightweight", pathjoin(vid_base, "model"), pathjoin(vid_base, "faces") </s> remove                 pathjoin(vid_base, "faces"), pathjoin(vid_base, "test_alignments.fsa"),
                iterations=1, extra_args="-wl"
 </s> add                 pathjoin(vid_base, "faces"), extra_args="-wl" </s> remove         print_ok("[+] Failed %i/%i tests." % (FAIL_COUNT, TEST_COUNT))
 </s> add         print_ok(f"[+] Failed {FAIL_COUNT}/{TEST_COUNT} tests.") </s> remove     _extract_args = "%s faceswap.py extract -i %s -o %s -D %s -A %s" % (
        py_exe, in_path, out_path, detector, aligner
    )
 </s> add     _extract_args = (f"{py_exe} faceswap.py extract -i {in_path} -o {out_path} -D {detector} "
                     f"-A {aligner}") </s> remove     stats = AMDStats(log_level)
 </s> add     stats = AMDStats(log_level=log_level)
            "Train lightweight model for 1 iterations WITHOUT WTL.",
              )
          )
  
          was_trained = run_test(
            "Train lightweight model for 1 iterations WITHOUT WTL.",
              train_args(
                  "lightweight", pathjoin(vid_base, "model"),
                  pathjoin(vid_base, "faces"), pathjoin(vid_base, "test_alignments.fsa")
              )
          )
keep keep keep keep replace keep keep keep keep keep
 Fix AMD Tests + docs 
tests/simple_tests.py