Add __version__ to init (#43)

Author: krfricke

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"
 ]

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