feat(ci): add dockerfile_fragments.py to ci/generate_code.sh to ensure code is in sync (#1635)

Author: jiridanek

ci/generate_code.sh

@@ -1,4 +1,5 @@
 #!/usr/bin/env bash
 set -Eeuxo pipefail
 
+python3 scripts/dockerfile_fragments.py
 bash scripts/sync-requirements-txt.sh

jupyter/pytorch+llmcompressor/ubi9-python-3.11/Dockerfile.cuda

@@ -30,11 +30,12 @@ USER 1001
 # Install micropipenv to deploy packages from Pipfile.lock
 RUN pip install --no-cache-dir -U "micropipenv[toml]"
 
-# Install the oc client
+# Install the oc client begin
 RUN curl -L https://mirror.openshift.com/pub/openshift-v4/$(uname -m)/clients/ocp/stable/openshift-client-linux.tar.gz \
         -o /tmp/openshift-client-linux.tar.gz && \
     tar -xzvf /tmp/openshift-client-linux.tar.gz oc && \
     rm -f /tmp/openshift-client-linux.tar.gz
+# Install the oc client end
 
 ####################
 # cuda-base        #

runtimes/pytorch+llmcompressor/ubi9-python-3.11/Dockerfile.cuda

@@ -17,11 +17,12 @@ USER 1001
 # Install micropipenv to deploy packages from Pipfile.lock
 RUN pip install --no-cache-dir -U "micropipenv[toml]"
 
-# Install the oc client
+# Install the oc client begin
 RUN curl -L https://mirror.openshift.com/pub/openshift-v4/$(uname -m)/clients/ocp/stable/openshift-client-linux.tar.gz \
         -o /tmp/openshift-client-linux.tar.gz && \
     tar -xzvf /tmp/openshift-client-linux.tar.gz oc && \
     rm -f /tmp/openshift-client-linux.tar.gz
+# Install the oc client end
 
 ####################
 # cuda-base        #

scripts/dockerfile_fragments.py

@@ -4,6 +4,10 @@
 import os
 import textwrap
 import pathlib
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from pyfakefs.fake_filesystem import FakeFilesystem
 
 ROOT_DIR = pathlib.Path(__file__).parent.parent
 
@@ -60,7 +64,7 @@ def blockinfile(filename: str | os.PathLike, contents: str, prefix: str | None =
         lines.extend(new_contents)
         lines.append(f"\n{end_marker}\n")
     else:
-        lines[begin : end + 1] = [f"{begin_marker}\n", *new_contents, f"\n{end_marker}\n"]
+        lines[begin: end + 1] = [f"{begin_marker}\n", *new_contents, f"\n{end_marker}\n"]
 
     if lines == original_lines:
         return
@@ -73,8 +77,6 @@ def blockinfile(filename: str | os.PathLike, contents: str, prefix: str | None =
 
 
 class TestBlockinfile:
-    from pyfakefs.fake_filesystem import FakeFilesystem
-
     def test_adding_new_block(self, fs: FakeFilesystem):
         fs.create_file("/config.txt", contents="hello\nworld")