the-algorithm-ml/common/test_device.py

"""Minimal test for device.

Mostly a test that this can be imported properly even tho moved.
"""
from unittest.mock import (
  patch,
)

import tml.common.device as device_utils


def test_device():
  with patch("tml.common.device.dist.init_process_group"):
    device = device_utils.setup_and_get_device(tf_ok=False)
  assert device.type == "cpu"
Twitter's Recommendation Algorithm - Heavy Ranker and TwHIN embeddings 2023-03-31 20:05:14 +02:00			`"""Minimal test for device.`

			`Mostly a test that this can be imported properly even tho moved.`
			`"""`
add & config mypy on common package Signed-off-by: wiseaidev <business@wiseai.dev> 2023-04-01 12:30:34 +02:00			`from unittest.mock import (`
			`patch,`
			`)`
Twitter's Recommendation Algorithm - Heavy Ranker and TwHIN embeddings 2023-03-31 20:05:14 +02:00
			`import tml.common.device as device_utils`


			`def test_device():`
			`with patch("tml.common.device.dist.init_process_group"):`
			`device = device_utils.setup_and_get_device(tf_ok=False)`
			`assert device.type == "cpu"`