From e10e95599e1a3cd0ea1dc2385dd74a7054a4cd97 Mon Sep 17 00:00:00 2001
From: gunchu <gunchu@Nikhils-MacBook-Pro.local>
Date: Wed, 29 Mar 2023 15:34:14 -0700
Subject: [PATCH] Fix `environment.yml` to include missing deps `execnb` and
 `nbformat` and add missing dep in `clean.py`. I was able to execute
 `clean.py` after these changes.

---
 environment.yml | 2 ++
 tools/clean.py  | 1 +
 2 files changed, 3 insertions(+)

diff --git a/environment.yml b/environment.yml
index cf41740..f597e1b 100644
--- a/environment.yml
+++ b/environment.yml
@@ -7,6 +7,8 @@ dependencies:
   - python>=3.6
   - pytorch>=1.6
   - torchvision
+  - execnb
+  - nbformat
   - pip
   - pip:
     - -r requirements.txt
diff --git a/tools/clean.py b/tools/clean.py
index 298e6b2..8417184 100755
--- a/tools/clean.py
+++ b/tools/clean.py
@@ -4,6 +4,7 @@ import nbformat
 from nbdev.export import *
 from nbdev.clean import *
 from fastcore.all import *
+from execnb.nbio import *
 
 _re_header = re.compile(r'^#+\s+\S+')
 _re_clean  = re.compile(r'^\s*#\s*clean\s*')