noqa on get_ipython

fix_#34
Gepetto · Aug 8, 2023 · cd3b0a0 · cd3b0a0
1 parent 5066e1e
commit cd3b0a0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -7,7 +7,6 @@ repos:
   rev: v0.0.282
   hooks:
   - id: ruff
-    exclude: gepetuto/magic.py
     args:
     - --fix
     - --exit-non-zero-on-fix

diff --git a/gepetuto/magic.py b/gepetuto/magic.py
@@ -12,8 +12,8 @@ class DoNotLoadMagics(Magics):
     def do_not_load_snippet(self, line):
         """Magic command to hide the snippet and let the student code by themself."""
         if DoNotLoadMagics.force_load:
-            get_ipython().run_line_magic("run", "-i " + line)
-            get_ipython().run_line_magic("load", line)  # noqa: F821
+            self.run("run", "-i " + line)
+            self.run("load", line)
 
     @line_magic
     def force_load(self, line):
@@ -28,8 +28,12 @@ def force_load(self, line):
     @line_magic
     def load_snippet(self, line):
         """Magic command to load and run the snippet, this only work on JupyterLab."""
-        get_ipython().run_line_magic("run", "-i " + line)
-        get_ipython().run_line_magic("load", line)
+        self.run("run", "-i " + line)
+        self.run("load", line)
+
+    def run(self, magic, line):
+        """Run magic command."""
+        get_ipython().run_line_magic(magic, line)  # noqa: F821
 
 
 ip = get_ipython()  # noqa: F821