gbrl 1.0.0.dev2__tar.gz → 1.0.0.dev3__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (50) hide show
  1. {gbrl-1.0.0.dev2/gbrl.egg-info → gbrl-1.0.0.dev3}/PKG-INFO +1 -1
  2. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/utils.cpp +1 -1
  3. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3/gbrl.egg-info}/PKG-INFO +1 -1
  4. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/pyproject.toml +1 -1
  5. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/setup.py +1 -8
  6. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/LICENSE +0 -0
  7. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/MANIFEST.in +0 -0
  8. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/README.md +0 -0
  9. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/fitter.cpp +0 -0
  10. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/fitter.h +0 -0
  11. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/gbrl.cpp +0 -0
  12. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/gbrl.h +0 -0
  13. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/gbrl_binding.cpp +0 -0
  14. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/loss.cpp +0 -0
  15. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/loss.h +0 -0
  16. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/main.cpp +0 -0
  17. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/math_ops.cpp +0 -0
  18. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/math_ops.h +0 -0
  19. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/node.cpp +0 -0
  20. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/node.h +0 -0
  21. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/optimizer.cpp +0 -0
  22. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/optimizer.h +0 -0
  23. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/predictor.cpp +0 -0
  24. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/predictor.h +0 -0
  25. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/scheduler.cpp +0 -0
  26. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/scheduler.h +0 -0
  27. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/split_candidate_generator.cpp +0 -0
  28. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/split_candidate_generator.h +0 -0
  29. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/types.cpp +0 -0
  30. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/types.h +0 -0
  31. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/utils.h +0 -0
  32. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_fitter.cu +0 -0
  33. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_fitter.h +0 -0
  34. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_loss.cu +0 -0
  35. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_loss.h +0 -0
  36. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_predictor.cu +0 -0
  37. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_predictor.h +0 -0
  38. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_preprocess.cu +0 -0
  39. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_preprocess.h +0 -0
  40. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_types.cu +0 -0
  41. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_types.h +0 -0
  42. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_utils.cu +0 -0
  43. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_utils.h +0 -0
  44. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl.egg-info/SOURCES.txt +0 -0
  45. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl.egg-info/dependency_links.txt +0 -0
  46. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl.egg-info/requires.txt +0 -0
  47. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl.egg-info/top_level.txt +0 -0
  48. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/setup.cfg +0 -0
  49. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/tests/test_gbt_multi.py +0 -0
  50. {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/tests/test_gbt_single.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: gbrl
3
- Version: 1.0.0.dev2
3
+ Version: 1.0.0.dev3
4
4
  Summary: Gradient Boosted Trees for RL
5
5
  Author-email: Benjamin Fuhrer <bfuhrer@nvidia.com>, Chen Tesslr <ctessler@nvidia.com>, Gal Dalal <galal@nvidia.com>
6
6
  License-File: LICENSE
@@ -21,7 +21,7 @@ std::string VectoString(const float* vec, const int vec_size){
21
21
 
22
22
  int binaryToDecimal(const BoolVector& binaryPath) {
23
23
  int decimal = 0;
24
- for (size_t i = binaryPath.size() - 1, j = 0; i >= 0; --i, ++j) {
24
+ for (size_t i = binaryPath.size() - 1, j = 0; i-- > 0; ++j) {
25
25
  decimal += binaryPath[i] * (1 << j);
26
26
  }
27
27
  return decimal + (1 << binaryPath.size()) - 1;
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: gbrl
3
- Version: 1.0.0.dev2
3
+ Version: 1.0.0.dev3
4
4
  Summary: Gradient Boosted Trees for RL
5
5
  Author-email: Benjamin Fuhrer <bfuhrer@nvidia.com>, Chen Tesslr <ctessler@nvidia.com>, Gal Dalal <galal@nvidia.com>
6
6
  License-File: LICENSE
@@ -8,7 +8,7 @@ build-backend = "setuptools.build_meta"
8
8
 
9
9
  [project]
10
10
  name = "gbrl"
11
- version = "1.0.0.dev2"
11
+ version = "1.0.0.dev3"
12
12
  description = "Gradient Boosted Trees for RL"
13
13
  authors = [
14
14
  {name = "Benjamin Fuhrer", email = "bfuhrer@nvidia.com"},
@@ -230,7 +230,7 @@ class CustomBuildExt(build_ext):
230
230
  Path(os.path.dirname(target)).mkdir(parents=True, exist_ok=True)
231
231
  command = [nvcc_path] + nvcc_compile_args + ['-c', source, '-o', target]
232
232
  print(' '.join(command))
233
- subprocess.check_call(command, shell=True)
233
+ subprocess.check_call(command)
234
234
  # Manually adding the compiled CUDA object files to be linked
235
235
  for idx, _ in enumerate(self.extensions):
236
236
  cuda_extras = [os.path.join(build_dir, f.replace('.cu', '.o')) for f in cuda_source_files]
@@ -240,13 +240,6 @@ class CustomBuildExt(build_ext):
240
240
  setup(
241
241
  name="gbrl",
242
242
  version="1.0.0",
243
- install_requires=["pybind11==2.11.1",
244
- "numpy",
245
- "torch",
246
- ],
247
- extras_require={
248
- 'sklearn': ['scikit_learn==1.2.2'], # Optional for unittests
249
- },
250
243
  packages=find_packages(include=["gbrl.*"], exclude=("tests*",)),
251
244
  package_data={
252
245
  'gbrl_cpp': ['*.so', '*.pyd'] # Include SO/PYD files
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes