Skip to content

Commit

Permalink
fix import
Browse files Browse the repository at this point in the history
  • Loading branch information
Difers committed Jan 7, 2024
1 parent 3a6dd30 commit b1ac26b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions test/ir/pir/translator/test_partial_send_translate.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,13 +14,13 @@

import unittest

import test_op_transcriber
import test_op_translator

import paddle
from paddle.base.layer_helper import LayerHelper


class TestCReduceMinOpTranscriber(test_op_transcriber.TestOpTranscriber):
class TestCPartialSendTranslator(test_op_translator.TestOpTranslator):
def append_op(self):
self.op_type = "partial_send"
x = paddle.ones(shape=(100, 2, 3), dtype='float32')
Expand Down

0 comments on commit b1ac26b

Please sign in to comment.