chore: fix e2e test Hf trainer searcher lengths (#9683)

e2e_tests/tests/nightly/test_pytorch2.py

@@ -15,7 +15,6 @@ def test_pytorch2_hf_language_modeling_distributed() -> None:
     config = conf.load_config(conf.hf_trainer_examples_path(f"{test_dir}/distributed.yaml"))
     config = conf.set_pt2_image(config)
     config = conf.set_slots_per_trial(config, 4)
-    config["searcher"]["max_length"]["batches"] = 50
 
     # Our hardware GPUs have only 16gb memory, lower memory use with smaller batches.
     config = conf.set_entrypoint(

examples/hf_trainer_api/hf_language_modeling/distributed.yaml

@@ -23,7 +23,7 @@ entrypoint: >-
   --dataset_config_name wikitext-2-raw-v1
   --do_train
   --do_eval
-  --num_train_epochs 5
+  --max_steps 100
   --logging_strategy steps
   --logging_steps 10
   --output_dir /tmp/test-clm

harness/.flake8

@@ -1,6 +1,6 @@
 # Note: This should be kept in sync with the .flake8 file in the root of the repo.
 [flake8]
-jobs = 8
+jobs = 4
 max-line-length = 100
 exclude =
     .git,