<reponame>karthik20122001/docker-python<filename>tests/test_pytorch_ignite.py
import unittest

from ignite.engine import Engine


class TestPytorchIgnite(unittest.TestCase):
    def test_engine(self):

        def update_fn(engine, batch):
            pass

        engine = Engine(update_fn)
        engine.run([0, 1, 2])

import unittest

from ignite.engine import Engine


class TestPytorchIgnite(unittest.TestCase):
    def test_engine(self):

        def update_fn(engine, batch):
            pass

        engine = Engine(update_fn)
        engine.run([0, 1, 2])