Add altair to requirements.txt and update Colab download section

Author: basilwong

.ci/docker/requirements.txt

@@ -50,6 +50,7 @@ onnxruntime
 evaluate
 accelerate>=0.20.1
 git+https://github.com/facebookresearch/mosaic.git
+altair
 
 importlib-metadata==6.8.0
 

beginner_source/mosaic_memory_profiling_tutorial.py

@@ -485,19 +485,16 @@ def run_training_ac(
 # Download Generated Files (Google Colab)
 # ----------------------------------------
 #
-# If running in Google Colab, you can download the generated snapshot
-# and profile files using the following code:
-#
-# .. code-block:: python
-#
-#    from google.colab import files
-#
-#    print("Downloading memory snapshots and profiles...")
-#    files.download('snapshot_baseline.pickle')
-#    files.download('snapshot_with_ac.pickle')
-#    files.download('profile_baseline.html')
-#    files.download('profile_with_ac.html')
+# If running in Google Colab, uncomment the following lines to download
+# the generated snapshot and profile files:
+
+# from google.colab import files
 #
+# print("Downloading memory snapshots and profiles...")
+# files.download('snapshot_baseline.pickle')
+# files.download('snapshot_with_ac.pickle')
+# files.download('profile_baseline.html')
+# files.download('profile_with_ac.html')
 
 ######################################################################
 # Results Interpretation: Activation Checkpointing