[Fix] notebook-to-mdx colab link creation workflow (#474)

Author: mishig25

.github/workflows/build_main_documentation.yml

@@ -155,15 +155,28 @@ jobs:
       - name: Convert notebooks to markdown files
         if: inputs.convert_notebooks
         run: |
+          initial_work_dir=$(pwd)
           cd ${{ env.doc_folder }}
           remote_url=$(git remote get-url origin)
           trimmed_url=${remote_url#https://}
           trimmed_url=${trimmed_url#github.com/}
           trimmed_url=${trimmed_url%.git}
           current_branch=$(git branch --show-current)
-          cd ..
+          cd $initial_work_dir
+
+          directory_string="${{ env.doc_folder }}"
+          if [[ "$directory_string" == *"/"* ]]; then
+              # directory_string contains "/"
+              remaining_part=${directory_string#*/}
+              if [[ -n $remaining_part ]]; then
+                  remaining_part="/$remaining_part"
+              fi
+          else
+              # directory_string does not contain "/"
+              remaining_part=""
+          fi
 
-          doc-builder notebook-to-mdx ${{ env.doc_folder }} --open_notebook_prefix https://colab.research.google.com/github/$trimmed_url/blob/$current_branch
+          doc-builder notebook-to-mdx ${{ env.doc_folder }} --open_notebook_prefix https://colab.research.google.com/github/$trimmed_url/blob/$current_branch$remaining_part
 
       - name: Make documentation
         shell: bash

.github/workflows/build_pr_documentation.yml

@@ -141,15 +141,28 @@ jobs:
       - name: Convert notebooks to markdown files
         if: inputs.convert_notebooks
         run: |
+          initial_work_dir=$(pwd)
           cd ${{ env.doc_folder }}
           remote_url=$(git remote get-url origin)
           trimmed_url=${remote_url#https://}
           trimmed_url=${trimmed_url#github.com/}
           trimmed_url=${trimmed_url%.git}
           current_branch=$(git branch --show-current)
-          cd ..
+          cd $initial_work_dir
+
+          directory_string="${{ env.doc_folder }}"
+          if [[ "$directory_string" == *"/"* ]]; then
+              # directory_string contains "/"
+              remaining_part=${directory_string#*/}
+              if [[ -n $remaining_part ]]; then
+                  remaining_part="/$remaining_part"
+              fi
+          else
+              # directory_string does not contain "/"
+              remaining_part=""
+          fi
 
-          doc-builder notebook-to-mdx ${{ env.doc_folder }} --open_notebook_prefix https://colab.research.google.com/github/$trimmed_url/blob/$current_branch
+          doc-builder notebook-to-mdx ${{ env.doc_folder }} --open_notebook_prefix https://colab.research.google.com/github/$trimmed_url/blob/$current_branch$remaining_part
 
       - name: Make documentation
         env: