ensure calls to .sample() are deterministic in CI

Author: afeld

.github/workflows/main.yml

@@ -27,7 +27,7 @@ jobs:
           # - hw_4.ipynb # has incomplete code
           - hw_5.ipynb
           - lecture_0.ipynb
-          # - lecture_1.ipynb # uses sample()
+          - lecture_1.ipynb
           - lecture_2.ipynb
           - lecture_3.ipynb
           - lecture_4.ipynb

scripts/diffable.py

@@ -16,6 +16,13 @@ def is_vid(cell):
     return text == "<IPython.core.display.Video object>"
 
 
+def fix_sample(line):
+    """Ensure calls to .sample() are deterministic by passing in a seed value
+
+    https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.sample.html"""
+    return re.sub(r"\.sample\((\d+)\)", r".sample(\1, random_state=1)", line)
+
+
 input_str = sys.stdin.read()
 notebook = json.loads(input_str)
 
@@ -34,6 +41,8 @@ def is_vid(cell):
     if cell["source"][0].startswith("!"):
         cell["outputs"] = []
 
+    cell["source"] = [fix_sample(line) for line in cell["source"]]
+
     # filter out pip upgrade warnings
     cell["outputs"] = [line for line in cell["outputs"] if not is_pip_upgrade_msg(line)]