• Reed Wanderman-Milne's avatar
    Return default strategy from get_distribution_strategy when given "off". · 9f9d07e9
    Reed Wanderman-Milne authored
    Before, it returned None. But almost every use of get_distribution_strategy() assumes an actual strategy is returned and crashes when None is returned. Returning the default strategy fixes these issues and is equivalent to using no strategy, as the default strategy is always in effect when no other strategy is used.
    
    PiperOrigin-RevId: 380951055
    9f9d07e9
distribute_utils.py 8.39 KB