Merge branch 'main' into ajkv/library-fix

Author: ajkv-google

training.md

@@ -6,7 +6,7 @@ This guide explains how to set up the environment and train the HSTU/DLRM models
 
 If you are developing on a TPU VM directly, use a virtual environment to avoid conflicts with the system-level Python packages.
 
-#### 1. Prerequisites
+### 1. Prerequisites
 Ensure you have **Python 3.11+** installed.
 ```bash
 python3 --version
@@ -41,7 +41,7 @@ python dlrm_experiment_test.py
 
 If you prefer not to manage a virtual environment or want to deploy this as a container, you can build a Docker image.
 
-## 1. Build the Image
+### 1. Create a Dockerfile
 Create a file named `Dockerfile` in the root of the repository:
 
 ```dockerfile
@@ -54,6 +54,9 @@ WORKDIR /app
 # This tells Python to look in /app for the 'recml' package
 ENV PYTHONPATH="${PYTHONPATH}:/app"
 
+# This tells Python to look in /app for the 'recml' package
+ENV PYTHONPATH="${PYTHONPATH}:/app"
+
 # Install system tools if needed (e.g., git)
 RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*