Add __version__ to init (#43)

ray-project · Jan 7, 2021 · 1f7e450 · 1f7e450
1 parent 20e28a7
commit 1f7e450
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/xgboost_ray/__init__.py b/xgboost_ray/__init__.py
@@ -3,7 +3,10 @@
     RayFileType, RayShardingMode, \
     Data, combine_data
 
+__version__ = "0.0.2"
+
 __all__ = [
-    "RayParams", "RayDMatrix", "RayDeviceQuantileDMatrix", "RayFileType",
-    "RayShardingMode", "Data", "combine_data", "train", "predict"
+    "__version__", "RayParams", "RayDMatrix", "RayDeviceQuantileDMatrix",
+    "RayFileType", "RayShardingMode", "Data", "combine_data", "train",
+    "predict"
 ]
diff --git a/xgboost_ray/main.py b/xgboost_ray/main.py
@@ -1,11 +1,11 @@
-import threading
 from typing import Tuple, Dict, Any, List, Optional, Callable, Union
 from dataclasses import dataclass, field
 
 import multiprocessing
 import os
 import pickle
 import time
+import threading
 
 import numpy as np