Skip to content

Commit

Permalink
linting
Browse files Browse the repository at this point in the history
  • Loading branch information
mohammadamint committed Dec 2, 2023
1 parent 8d39969 commit dd13fc6
Showing 1 changed file with 10 additions and 6 deletions.
16 changes: 10 additions & 6 deletions docs/notebooks_convert.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,11 @@
import nbformat
import os

path = os.path.abspath(os.path.join(os.path.dirname(__file__),))
path = os.path.abspath(
os.path.join(
os.path.dirname(__file__),
)
)


def convert_notebook_to_rst(input_file, output_dir, file_name):
Expand All @@ -20,7 +24,7 @@ def convert_notebook_to_rst(input_file, output_dir, file_name):
if not os.path.exists(output_dir):
os.mkdir(output_dir)

output_file = os.path.join(output_dir, f"{{file_name}.rst")
output_file = os.path.join(output_dir, f"{file_name}.rst")

# Write RST content to the output file
with open(output_file, "w", encoding="utf-8") as f:
Expand All @@ -31,13 +35,12 @@ def convert_notebook_to_rst(input_file, output_dir, file_name):
# Copy images to the output folder
image_dir = output_dir
for image_filename, image_data in resources["outputs"].items():
image_path = os.path,join(output_dir, image_filename)
image_path = os.path.join(output_dir, image_filename)
with open(image_path, "wb") as img_file:
img_file.write(image_data)


def get_all_ipynb_files(examples_path="notebooks"):

all_files = os.listdir(examples_path)
return [
(os.path.join(path, examples_path, file), file.split(".ipynb")[0])
Expand All @@ -47,8 +50,9 @@ def get_all_ipynb_files(examples_path="notebooks"):


if __name__ == "__main__":

files = get_all_ipynb_files()

for file, folder in files:
convert_notebook_to_rst(file, os.path.join(path, source, examples,folder), folder)
convert_notebook_to_rst(
file, os.path.join(path, "source", "examples", folder), folder
)

0 comments on commit dd13fc6

Please sign in to comment.