def tf_distribution()

in sage_maker_magic/sage_maker_kernel/kernelmagics.py [0:0]

• 12 lines of code
• 1 McCabe index (conditional complexity)

    def tf_distribution(self, choise):
        distribution = {
            "parameter_server": {"parameter_server": {"enabled": True}},
            "horovod": {
                "mpi": {
                    "enabled": True,
                    "processes_per_host": self.args['mpi_processes_per_host'],
                    "custom_mpi_options": self.args['mpi_custom_mpi_options']
                },
            }
        }
        return distribution.get(choise, None)