fix bug in dataset/_init_.py

WangGodder · Apr 8, 2020 · 4c7f450 · 4c7f450
1 parent 5f47b22
commit 4c7f450
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/torchcmh/dataset/base/__init__.py b/torchcmh/dataset/base/__init__.py
@@ -14,6 +14,5 @@
 
 
 __all__ = ['CrossModalTrainBase', 'CrossModalValidBase', 'CrossModalSingleTrain',
-           'CrossModalPairwiseTrain', 'CrossModalTripletTrain', 'CrossModalQuadrupletTrain',
-           'CrossModalTripletRankingTrain']
+           'CrossModalPairwiseTrain', 'CrossModalTripletTrain', 'CrossModalQuadrupletTrain']