barian0517/APEX_AIMBOT
1
0
Fork 0
mirror of https://github.com/RootKit-Org/AI-Aimbot.git synced 2025-06-21 02:41:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

bugfix: fix issue 'cannot import name is_notebook' (#41)

Browse Source
This commit is contained in:
Fabio Monreal 2022-10-30 23:34:12 -03:00 committed by GitHub
parent 65cb7a2111
commit 64a3927762
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils/general.py
View File

@ -27,6 +27,7 @@ from typing import Optional
from zipfile import ZipFile
import cv2
import IPython
import numpy as np
import pandas as pd
import pkg_resources as pkg
@ -137,6 +138,11 @@ def user_config_dir(dir='Ultralytics', env_var='YOLOV5_CONFIG_DIR'):
path.mkdir(exist_ok=True) # make if required
return path
def is_notebook():
# Is environment a Jupyter notebook? Verified on Colab, Jupyterlab, Kaggle, Paperspace
ipython_type = str(type(IPython.get_ipython()))
return 'colab' in ipython_type or 'zmqshell' in ipython_type
CONFIG_DIR = user_config_dir() # Ultralytics settings dir