Needs pypi package

Author: rithikanarayan

ci/input_files/build.yaml.tpl

@@ -212,9 +212,8 @@ update-layer-versions-docs:
     project: DataDog/serverless-ci
   rules:
     - if: '$CI_COMMIT_TAG =~ /^v.*/'
-  needs: {{ range $runtime := (ds "runtimes").runtimes }}
-    - publish-layer-prod ({{ $runtime.name }}-{{ $runtime.arch}})
-    {{- end }}
+  needs:
+    - publish-pypi-package
   variables:
     RUN_LAMBDA_LAYER_DOCUMENTATION: "true"
     RUN_LAMBDA_DATADOG_CI: "true"