andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Add nbconvert by default

Browse source
This commit is contained in:
Andras Schmelczer 2022-06-25 18:33:17 +02:00
parent f423cd716f
commit 844fae4ef5
No known key found for this signature in database
GPG key ID: 0EA1BC97D0AB076E
2 changed files with 8 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

1
setup.cfg
View file

@ -41,6 +41,7 @@ install_requires =
fastapi >= 0.70.0 fastapi >= 0.70.0
plotly >= 5.8.0 plotly >= 5.8.0
dash >= 2.4.0 dash >= 2.4.0
nbconvert >= 6.5.0
uvicorn[standard] >= 0.18.0 uvicorn[standard] >= 0.18.0
watchdog >= 2.1.0 watchdog >= 2.1.0
pymongo >= 4.0.0 pymongo >= 4.0.0

6
src/great_ai/__main__.py
View file

@ -116,18 +116,14 @@ def main() -> None:
def get_script_name(file_name_argument: str) -> str: def get_script_name(file_name_argument: str) -> str:
if file_name_argument.endswith(".ipynb"): if file_name_argument.endswith(".ipynb"):
logger.info("Converting notebook to Python script") logger.info("Converting notebook to Python script")
try:
from nbconvert import PythonExporter from nbconvert import PythonExporter
exporter = PythonExporter() exporter = PythonExporter()
content, _ = exporter.from_filename(file_name_argument) content, _ = exporter.from_filename(file_name_argument)
file_name_argument = get_script_name_of_notebook(file_name_argument) file_name_argument = get_script_name_of_notebook(file_name_argument)
with open(file_name_argument, "w", encoding="utf-8") as f: with open(file_name_argument, "w", encoding="utf-8") as f:
f.write(content) f.write(content)
except ImportError:
raise ImportError(
"Install `nbconvert` to be able to use Jupyter notebook files or use a regular Python file instead"
)
return re.sub(r"\.(py|ipynb)$", "", file_name_argument) return re.sub(r"\.(py|ipynb)$", "", file_name_argument)