fix lint

liuwenran · Aug 21, 2023 · 29d800a · 29d800a
1 parent 66792c1
commit 29d800a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/mmagic/models/editors/stable_diffusion/stable_diffusion.py b/mmagic/models/editors/stable_diffusion/stable_diffusion.py
@@ -612,12 +612,13 @@ def test_step(self, data: dict) -> SampleList:
         data_samples = data['data_samples']
         prompt = data_samples.prompt
 
-        output = self.infer(prompt, return_type='tensor', height=256, width=256)
+        output = self.infer(
+            prompt, return_type='tensor', height=256, width=256)
         samples = output['samples']
 
         samples = self.data_preprocessor.destruct(samples, data_samples)
-        # gt_img = self.data_preprocessor.destruct(data['inputs'], data_samples)
-        gt_img = ['temp']
+        gt_img = self.data_preprocessor.destruct(data['inputs'], data_samples)
+        # gt_img = ['temp']
 
         out_data_sample = DataSample(
             fake_img=samples, gt_img=gt_img, prompt=prompt)

diff --git a/mmagic/structures/data_sample.py b/mmagic/structures/data_sample.py
@@ -244,9 +244,7 @@ def set_gt_label(
             self.gt_label = label
         return self
 
-    def set_gt_prompt(
-        self, value: Union[str, Sequence[str]]
-    ) -> 'DataSample':
+    def set_gt_prompt(self, value: Union[str, Sequence[str]]) -> 'DataSample':
         """Set label of ``gt_label``."""
         self.prompt = value
         return self