From 2b4b516dab647fa0e409808864d6d6edc04648f2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gustavo=20Eduardo=20Vargas=20N=C3=BA=C3=B1ez?=
<gvargas@ETS.ES>
Date: Mon, 8 Feb 2021 15:50:01 +0100
Subject: [PATCH] fix validate_nb for compatibility with google colab
---
ipynb/utils.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/ipynb/utils.py b/ipynb/utils.py
index b2414f8..6c2ee46 100644
--- a/ipynb/utils.py
+++ b/ipynb/utils.py
@@ -34,10 +34,11 @@ def validate_nb(nb):
if nb['nbformat'] != 4:
return False
- language_name = (nb.get('metadata', {})
- .get('kernelspec', {})
- .get('language', '').lower())
- return language_name == 'python'
+ kernelspec = nb.get('metadata', {}).get('kernelspec', {})
+ lang_name1 = kernelspec.get('language', '').lower()
+ lang_name2 = kernelspec.get('name', '').lower()
+ language_names = [lang_name1, lang_name2]
+ return any(x in language_names for x in ['python3', 'python'])
def filter_ast(module_ast):