From 4e94706bc158dbe7606b9c88bd1dbb737a8c618d Mon Sep 17 00:00:00 2001
From: Brilliant Hanabi <moehanabichan@gmail.com>
Date: Fri, 22 Nov 2024 22:20:35 +0800
Subject: [PATCH] fix

---
 tests/ops/test_ops_torch_nn.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/tests/ops/test_ops_torch_nn.py b/tests/ops/test_ops_torch_nn.py
index c909713..22158af 100644
--- a/tests/ops/test_ops_torch_nn.py
+++ b/tests/ops/test_ops_torch_nn.py
@@ -1,8 +1,9 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
 import os
 import unittest
 
+os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
+os.environ['TL_BACKEND'] = 'torch'
+
 import numpy as np
 import torch
 import torch.nn.functional as F
@@ -11,8 +12,6 @@
 from tensorlayerx.backend.ops.torch_nn import *
 from tests.utils import CustomTestCase
 
-os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
-
 
 class TestPaddingFormat(CustomTestCase):