Don't wrap None into List.

gmalivenko · Jun 27, 2019 · c27f1a1 · c27f1a1
1 parent 52c4160
commit c27f1a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch2keras/converter.py b/pytorch2keras/converter.py
@@ -41,7 +41,7 @@ def pytorch_to_keras(
     if input_shapes:
         logger.warning('Custom shapes isn\'t supported now.')
 
-    if not isinstance(input_shapes, list):
+    if input_shapes and not isinstance(input_shapes, list):
         input_shapes = [input_shapes]
 
     if not isinstance(args, list):