mirror of
https://github.com/Shawn-Shan/fawkes.git
synced 2024-12-22 07:09:33 +05:30
Use ==/!= to compare constant literals (str, bytes, int, float, tuple)
Avoid [SyntaxWarnings on Python >= 3.8](https://docs.python.org/3/whatsnew/3.8.html#porting-to-python-3-8). % `python3.8` ``` >>> 0 is 0 <stdin>:1: SyntaxWarning: "is" with a literal. Did you mean "=="? ```
This commit is contained in:
parent
0180c98e20
commit
b61dfadb97
@ -1,3 +1,4 @@
|
||||
import errno
|
||||
import glob
|
||||
import gzip
|
||||
import json
|
||||
@ -243,9 +244,9 @@ def fix_gpu_memory(mem_fraction=1):
|
||||
def preprocess(X, method):
|
||||
assert method in {'raw', 'imagenet', 'inception', 'mnist'}
|
||||
|
||||
if method is 'raw':
|
||||
if method == 'raw':
|
||||
pass
|
||||
elif method is 'imagenet':
|
||||
elif method == 'imagenet':
|
||||
X = imagenet_preprocessing(X)
|
||||
else:
|
||||
raise Exception('unknown method %s' % method)
|
||||
@ -256,9 +257,9 @@ def preprocess(X, method):
|
||||
def reverse_preprocess(X, method):
|
||||
assert method in {'raw', 'imagenet', 'inception', 'mnist'}
|
||||
|
||||
if method is 'raw':
|
||||
if method == 'raw':
|
||||
pass
|
||||
elif method is 'imagenet':
|
||||
elif method == 'imagenet':
|
||||
X = imagenet_reverse_preprocessing(X)
|
||||
else:
|
||||
raise Exception('unknown method %s' % method)
|
||||
|
Loading…
Reference in New Issue
Block a user