<filename>tests/common_setup.py
# Copyright (c) 2019 NVIDIA Corporation
import unittest
import os
from .context import nemo
# from nemo.backends.pytorch.nm import TrainableNM
import torch.nn as nn


class NeMoUnitTest(unittest.TestCase):

    def setUp(self) -> None:
        nemo.core.neural_factory.NeuralModuleFactory.reset_default_factory()

<filename>tests/common_setup.py
# Copyright (c) 2019 NVIDIA Corporation
import unittest
import os
from .context import nemo
# from nemo.backends.pytorch.nm import TrainableNM
import torch.nn as nn


class NeMoUnitTest(unittest.TestCase):

    def setUp(self) -> None:
        nemo.core.neural_factory.NeuralModuleFactory.reset_default_factory()

# Copyright (c) 2019 NVIDIA Corporation # from nemo.backends.pytorch.nm import TrainableNM