rlaphoenix / VSGAN

Consider using f-strings PYL-C0209
Performance
Minor
8 months ago2 years old
Formatting a regular string which could be a f-string
 89            "pixel_shuffle": blocks.pixel_shuffle_block
 90        }.get(self.upsampler)
 91        if upsample_block is None:
 92            raise NotImplementedError("Upsample mode [%s] is not found" % self.upsampler) 93
 94        if self.scale == 3:
 95            upsample_blocks = upsample_block(
Formatting a regular string which could be a f-string
 37            "BD": BicubicUpsample(scale_factor=scale)
 38        }.get(degradation)
 39        if self.upsample_func is None:
 40            raise NotImplementedError("Degradation [%s] is not recognized" % degradation) 41
 42        self.fnet = FNet(in_nc)
 43        self.srnet = SRNet(in_nc, out_nc, nf, nb, self.upsample_func)
Formatting a regular string which could be a f-string
40        if device == "cpu!":
41            device = "cpu"
42        if device != "cpu" and not torch.cuda.is_available():
43            raise EnvironmentError("Either NVIDIA CUDA or the device (%s) isn't available." % device)44
45        self.clip: vs.VideoNode = clip
46        self._device: torch.device = torch.device(device)