egg/zoo/language_bottleneck/intervention.py
egg/zoo/objects_game/util.py

egg/zoo/language_bottleneck/guess_number/train.py
egg/zoo/objects_game/train.py