Skip to content

Commit c7f6683

Browse files
authored
Fix @neptune_db_only magics decorator (#543)
1 parent 8bdef5c commit c7f6683

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/graph_notebook/decorators/decorators.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,8 @@ def use_magic_variables(*args, **kwargs):
136136

137137
def neptune_db_only(func):
138138
@functools.wraps(func)
139-
def check_neptune_db(self, *args, **kwargs):
139+
def check_neptune_db(*args, **kwargs):
140+
self = args[0]
140141
if not hasattr(self.graph_notebook_config, 'neptune_service'):
141142
return func(*args, **kwargs)
142143
else:

0 commit comments

Comments
 (0)