  
          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")
              )
          )
  
      if was_trained:
          run_test(
 
</s> remove             "Train lightweight model for 5 iterations WITHOUT WTL.",
 </s> add             "Train lightweight model for 1 iterations WITHOUT WTL.", </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     stats = AMDStats(log_level)
 </s> add     stats = AMDStats(log_level=log_level) </s> remove         details = [json.loads(d.details.decode("utf-8"))
                   for d in self._all_devices if d.details]
 </s> add         details = []
        for dev in self._all_devices:
            if dev.details:
                details.append(json.loads(dev.details.decode("utf-8")))
            else:
                details.append(dict(vendor=dev.id.decode("utf-8"),
                                    name=dev.description.decode("utf-8"),
                                    globalMemSize=4 * 1024 * 1024 * 1024))  # 4GB dummy ram </s> remove     logger.verbose("Setting Keras Backend to PlaidML")
 </s> add     logger.verbose("Setting Keras Backend to PlaidML")  # type:ignore
                "lightweight", pathjoin(vid_base, "model"), pathjoin(vid_base, "faces")
  
          was_trained = run_test(
              "Train lightweight model for 5 iterations WITHOUT WTL.",
              train_args(
                "lightweight", pathjoin(vid_base, "model"), pathjoin(vid_base, "faces")
                "lightweight", pathjoin(vid_base, "model"), pathjoin(vid_base, "faces")
              )
          )
  
      if was_trained:
          run_test(
keep keep keep keep replace replace keep keep keep keep keep
 Fix AMD Tests + docs 
tests/simple_tests.py