nemo-evaluator-launcher 0.1.0rc7__tar.gz → 0.1.0rc9__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.

Potentially problematic release.


This version of nemo-evaluator-launcher might be problematic. Click here for more details.

Files changed (64) hide show
  1. nemo_evaluator_launcher-0.1.0rc9/PKG-INFO +494 -0
  2. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/pyproject.toml +29 -4
  3. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/api/functional.py +38 -1
  4. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/executors/slurm/executor.py +15 -2
  5. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/package_info.py +4 -1
  6. nemo_evaluator_launcher-0.1.0rc9/src/nemo_evaluator_launcher.egg-info/PKG-INFO +494 -0
  7. nemo_evaluator_launcher-0.1.0rc7/PKG-INFO +0 -35
  8. nemo_evaluator_launcher-0.1.0rc7/src/nemo_evaluator_launcher.egg-info/PKG-INFO +0 -35
  9. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/LICENSE +0 -0
  10. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/README.md +0 -0
  11. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/setup.cfg +0 -0
  12. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/__init__.py +0 -0
  13. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/api/__init__.py +0 -0
  14. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/api/types.py +0 -0
  15. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/api/utils.py +0 -0
  16. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/cli/__init__.py +0 -0
  17. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/cli/export.py +0 -0
  18. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/cli/info.py +0 -0
  19. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/cli/kill.py +0 -0
  20. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/cli/ls_runs.py +0 -0
  21. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/cli/ls_tasks.py +0 -0
  22. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/cli/main.py +0 -0
  23. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/cli/run.py +0 -0
  24. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/cli/status.py +0 -0
  25. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/cli/version.py +0 -0
  26. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/common/__init__.py +0 -0
  27. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/common/execdb.py +0 -0
  28. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/common/helpers.py +0 -0
  29. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/common/logging_utils.py +0 -0
  30. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/common/mapping.py +0 -0
  31. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/configs/__init__.py +0 -0
  32. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/configs/default.yaml +0 -0
  33. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/configs/deployment/nim.yaml +0 -0
  34. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/configs/deployment/none.yaml +0 -0
  35. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/configs/deployment/sglang.yaml +0 -0
  36. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/configs/deployment/vllm.yaml +0 -0
  37. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/configs/execution/lepton/default.yaml +0 -0
  38. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/configs/execution/local.yaml +0 -0
  39. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/configs/execution/slurm/default.yaml +0 -0
  40. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/executors/__init__.py +0 -0
  41. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/executors/base.py +0 -0
  42. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/executors/lepton/__init__.py +0 -0
  43. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/executors/lepton/deployment_helpers.py +0 -0
  44. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/executors/lepton/executor.py +0 -0
  45. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/executors/lepton/job_helpers.py +0 -0
  46. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/executors/local/__init__.py +0 -0
  47. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/executors/local/executor.py +0 -0
  48. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/executors/local/run.template.sh +0 -0
  49. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/executors/registry.py +0 -0
  50. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/executors/slurm/__init__.py +0 -0
  51. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/exporters/__init__.py +0 -0
  52. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/exporters/base.py +0 -0
  53. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/exporters/gsheets.py +0 -0
  54. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/exporters/local.py +0 -0
  55. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/exporters/mlflow.py +0 -0
  56. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/exporters/registry.py +0 -0
  57. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/exporters/utils.py +0 -0
  58. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/exporters/wandb.py +0 -0
  59. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher/resources/mapping.toml +0 -0
  60. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher.egg-info/SOURCES.txt +0 -0
  61. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher.egg-info/dependency_links.txt +0 -0
  62. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher.egg-info/entry_points.txt +0 -0
  63. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher.egg-info/requires.txt +0 -0
  64. {nemo_evaluator_launcher-0.1.0rc7 → nemo_evaluator_launcher-0.1.0rc9}/src/nemo_evaluator_launcher.egg-info/top_level.txt +0 -0
@@ -0,0 +1,494 @@
1
+ Metadata-Version: 2.4
2
+ Name: nemo-evaluator-launcher
3
+ Version: 0.1.0rc9
4
+ Summary: Launcher for the evaluations provided by NeMo Evaluator containers with different runtime backends
5
+ Author: NVIDIA
6
+ Author-email: nemo-toolkit@nvidia.com
7
+ Maintainer: NVIDIA
8
+ Maintainer-email: nemo-toolkit@nvidia.com
9
+ License:
10
+ Apache License
11
+ Version 2.0, January 2004
12
+ http://www.apache.org/licenses/
13
+
14
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
15
+
16
+ 1. Definitions.
17
+
18
+ "License" shall mean the terms and conditions for use, reproduction,
19
+ and distribution as defined by Sections 1 through 9 of this document.
20
+
21
+ "Licensor" shall mean the copyright owner or entity authorized by
22
+ the copyright owner that is granting the License.
23
+
24
+ "Legal Entity" shall mean the union of the acting entity and all
25
+ other entities that control, are controlled by, or are under common
26
+ control with that entity. For the purposes of this definition,
27
+ "control" means (i) the power, direct or indirect, to cause the
28
+ direction or management of such entity, whether by contract or
29
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
30
+ outstanding shares, or (iii) beneficial ownership of such entity.
31
+
32
+ "You" (or "Your") shall mean an individual or Legal Entity
33
+ exercising permissions granted by this License.
34
+
35
+ "Source" form shall mean the preferred form for making modifications,
36
+ including but not limited to software source code, documentation
37
+ source, and configuration files.
38
+
39
+ "Object" form shall mean any form resulting from mechanical
40
+ transformation or translation of a Source form, including but
41
+ not limited to compiled object code, generated documentation,
42
+ and conversions to other media types.
43
+
44
+ "Work" shall mean the work of authorship, whether in Source or
45
+ Object form, made available under the License, as indicated by a
46
+ copyright notice that is included in or attached to the work
47
+ (an example is provided in the Appendix below).
48
+
49
+ "Derivative Works" shall mean any work, whether in Source or Object
50
+ form, that is based on (or derived from) the Work and for which the
51
+ editorial revisions, annotations, elaborations, or other modifications
52
+ represent, as a whole, an original work of authorship. For the purposes
53
+ of this License, Derivative Works shall not include works that remain
54
+ separable from, or merely link (or bind by name) to the interfaces of,
55
+ the Work and Derivative Works thereof.
56
+
57
+ "Contribution" shall mean any work of authorship, including
58
+ the original version of the Work and any modifications or additions
59
+ to that Work or Derivative Works thereof, that is intentionally
60
+ submitted to Licensor for inclusion in the Work by the copyright owner
61
+ or by an individual or Legal Entity authorized to submit on behalf of
62
+ the copyright owner. For the purposes of this definition, "submitted"
63
+ means any form of electronic, verbal, or written communication sent
64
+ to the Licensor or its representatives, including but not limited to
65
+ communication on electronic mailing lists, source code control systems,
66
+ and issue tracking systems that are managed by, or on behalf of, the
67
+ Licensor for the purpose of discussing and improving the Work, but
68
+ excluding communication that is conspicuously marked or otherwise
69
+ designated in writing by the copyright owner as "Not a Contribution."
70
+
71
+ "Contributor" shall mean Licensor and any individual or Legal Entity
72
+ on behalf of whom a Contribution has been received by Licensor and
73
+ subsequently incorporated within the Work.
74
+
75
+ 2. Grant of Copyright License. Subject to the terms and conditions of
76
+ this License, each Contributor hereby grants to You a perpetual,
77
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
78
+ copyright license to reproduce, prepare Derivative Works of,
79
+ publicly display, publicly perform, sublicense, and distribute the
80
+ Work and such Derivative Works in Source or Object form.
81
+
82
+ 3. Grant of Patent License. Subject to the terms and conditions of
83
+ this License, each Contributor hereby grants to You a perpetual,
84
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
85
+ (except as stated in this section) patent license to make, have made,
86
+ use, offer to sell, sell, import, and otherwise transfer the Work,
87
+ where such license applies only to those patent claims licensable
88
+ by such Contributor that are necessarily infringed by their
89
+ Contribution(s) alone or by combination of their Contribution(s)
90
+ with the Work to which such Contribution(s) was submitted. If You
91
+ institute patent litigation against any entity (including a
92
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
93
+ or a Contribution incorporated within the Work constitutes direct
94
+ or contributory patent infringement, then any patent licenses
95
+ granted to You under this License for that Work shall terminate
96
+ as of the date such litigation is filed.
97
+
98
+ 4. Redistribution. You may reproduce and distribute copies of the
99
+ Work or Derivative Works thereof in any medium, with or without
100
+ modifications, and in Source or Object form, provided that You
101
+ meet the following conditions:
102
+
103
+ (a) You must give any other recipients of the Work or
104
+ Derivative Works a copy of this License; and
105
+
106
+ (b) You must cause any modified files to carry prominent notices
107
+ stating that You changed the files; and
108
+
109
+ (c) You must retain, in the Source form of any Derivative Works
110
+ that You distribute, all copyright, patent, trademark, and
111
+ attribution notices from the Source form of the Work,
112
+ excluding those notices that do not pertain to any part of
113
+ the Derivative Works; and
114
+
115
+ (d) If the Work includes a "NOTICE" text file as part of its
116
+ distribution, then any Derivative Works that You distribute must
117
+ include a readable copy of the attribution notices contained
118
+ within such NOTICE file, excluding those notices that do not
119
+ pertain to any part of the Derivative Works, in at least one
120
+ of the following places: within a NOTICE text file distributed
121
+ as part of the Derivative Works; within the Source form or
122
+ documentation, if provided along with the Derivative Works; or,
123
+ within a display generated by the Derivative Works, if and
124
+ wherever such third-party notices normally appear. The contents
125
+ of the NOTICE file are for informational purposes only and
126
+ do not modify the License. You may add Your own attribution
127
+ notices within Derivative Works that You distribute, alongside
128
+ or as an addendum to the NOTICE text from the Work, provided
129
+ that such additional attribution notices cannot be construed
130
+ as modifying the License.
131
+
132
+ You may add Your own copyright statement to Your modifications and
133
+ may provide additional or different license terms and conditions
134
+ for use, reproduction, or distribution of Your modifications, or
135
+ for any such Derivative Works as a whole, provided Your use,
136
+ reproduction, and distribution of the Work otherwise complies with
137
+ the conditions stated in this License.
138
+
139
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
140
+ any Contribution intentionally submitted for inclusion in the Work
141
+ by You to the Licensor shall be under the terms and conditions of
142
+ this License, without any additional terms or conditions.
143
+ Notwithstanding the above, nothing herein shall supersede or modify
144
+ the terms of any separate license agreement you may have executed
145
+ with Licensor regarding such Contributions.
146
+
147
+ 6. Trademarks. This License does not grant permission to use the trade
148
+ names, trademarks, service marks, or product names of the Licensor,
149
+ except as required for reasonable and customary use in describing the
150
+ origin of the Work and reproducing the content of the NOTICE file.
151
+
152
+ 7. Disclaimer of Warranty. Unless required by applicable law or
153
+ agreed to in writing, Licensor provides the Work (and each
154
+ Contributor provides its Contributions) on an "AS IS" BASIS,
155
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
156
+ implied, including, without limitation, any warranties or conditions
157
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
158
+ PARTICULAR PURPOSE. You are solely responsible for determining the
159
+ appropriateness of using or redistributing the Work and assume any
160
+ risks associated with Your exercise of permissions under this License.
161
+
162
+ 8. Limitation of Liability. In no event and under no legal theory,
163
+ whether in tort (including negligence), contract, or otherwise,
164
+ unless required by applicable law (such as deliberate and grossly
165
+ negligent acts) or agreed to in writing, shall any Contributor be
166
+ liable to You for damages, including any direct, indirect, special,
167
+ incidental, or consequential damages of any character arising as a
168
+ result of this License or out of the use or inability to use the
169
+ Work (including but not limited to damages for loss of goodwill,
170
+ work stoppage, computer failure or malfunction, or any and all
171
+ other commercial damages or losses), even if such Contributor
172
+ has been advised of the possibility of such damages.
173
+
174
+ 9. Accepting Warranty or Additional Liability. While redistributing
175
+ the Work or Derivative Works thereof, You may choose to offer,
176
+ and charge a fee for, acceptance of support, warranty, indemnity,
177
+ or other liability obligations and/or rights consistent with this
178
+ License. However, in accepting such obligations, You may act only
179
+ on Your own behalf and on Your sole responsibility, not on behalf
180
+ of any other Contributor, and only if You agree to indemnify,
181
+ defend, and hold each Contributor harmless for any liability
182
+ incurred by, or claims asserted against, such Contributor by reason
183
+ of your accepting any such warranty or additional liability.
184
+
185
+ END OF TERMS AND CONDITIONS
186
+
187
+ Copyright 2021 NVIDIA Corporation
188
+
189
+ Licensed under the Apache License, Version 2.0 (the "License");
190
+ you may not use this file except in compliance with the License.
191
+ You may obtain a copy of the License at
192
+
193
+ http://www.apache.org/licenses/LICENSE-2.0
194
+
195
+ Unless required by applicable law or agreed to in writing, software
196
+ distributed under the License is distributed on an "AS IS" BASIS,
197
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
198
+ See the License for the specific language governing permissions and
199
+ limitations under the License.
200
+
201
+
202
+ PORTIONS LICENSED AS FOLLOWS
203
+
204
+ > tools/pytorch-quantization/examples/torchvision/models/classification/resnet.py
205
+
206
+ BSD 3-Clause License
207
+
208
+ Copyright (c) Soumith Chintala 2016,
209
+ All rights reserved.
210
+
211
+ Redistribution and use in source and binary forms, with or without
212
+ modification, are permitted provided that the following conditions are met:
213
+
214
+ * Redistributions of source code must retain the above copyright notice, this
215
+ list of conditions and the following disclaimer.
216
+
217
+ * Redistributions in binary form must reproduce the above copyright notice,
218
+ this list of conditions and the following disclaimer in the documentation
219
+ and/or other materials provided with the distribution.
220
+
221
+ * Neither the name of the copyright holder nor the names of its
222
+ contributors may be used to endorse or promote products derived from
223
+ this software without specific prior written permission.
224
+
225
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
226
+ AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
227
+ IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
228
+ DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
229
+ FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
230
+ DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
231
+ SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
232
+ CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
233
+ OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
234
+ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
235
+
236
+ > samples/common/windows/getopt.c
237
+
238
+ Copyright (c) 2002 Todd C. Miller <Todd.Miller@courtesan.com>
239
+
240
+ Permission to use, copy, modify, and distribute this software for any
241
+ purpose with or without fee is hereby granted, provided that the above
242
+ copyright notice and this permission notice appear in all copies.
243
+
244
+ THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
245
+ WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
246
+ MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
247
+ ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
248
+ WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
249
+ ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
250
+ OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
251
+
252
+ Sponsored in part by the Defense Advanced Research Projects
253
+ Agency (DARPA) and Air Force Research Laboratory, Air Force
254
+ Materiel Command, USAF, under agreement number F39502-99-1-0512.
255
+
256
+
257
+ Copyright (c) 2000 The NetBSD Foundation, Inc.
258
+ All rights reserved.
259
+
260
+ This code is derived from software contributed to The NetBSD Foundation
261
+ by Dieter Baron and Thomas Klausner.
262
+
263
+ Redistribution and use in source and binary forms, with or without
264
+ modification, are permitted provided that the following conditions
265
+ are met:
266
+ 1. Redistributions of source code must retain the above copyright
267
+ notice, this list of conditions and the following disclaimer.
268
+ 2. Redistributions in binary form must reproduce the above copyright
269
+ notice, this list of conditions and the following disclaimer in the
270
+ documentation and/or other materials provided with the distribution.
271
+
272
+ THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
273
+ ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
274
+ TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
275
+ PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
276
+ BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
277
+ CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
278
+ SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
279
+ INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
280
+ CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
281
+ ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
282
+ POSSIBILITY OF SUCH DAMAGE.
283
+ - Copyright (c) 2002 Todd C. Miller <Todd.Miller@courtesan.com>
284
+ - Copyright (c) 2000 The NetBSD Foundation, Inc.
285
+
286
+
287
+ > parsers/common/ieee_half.h
288
+ > samples/common/half.h
289
+ > third_party/ieee/half.h
290
+
291
+ The MIT License
292
+
293
+ Copyright (c) 2012-2017 Christian Rau <rauy@users.sourceforge.net>
294
+
295
+ Permission is hereby granted, free of charge, to any person obtaining a
296
+ copy of this software and associated documentation files (the "Software"),
297
+ to deal in the Software without restriction, including without limitation
298
+ the rights to use, copy, modify, merge, publish, distribute, sublicense,
299
+ and/or sell copies of the Software, and to permit persons to whom the
300
+ Software is furnished to do so, subject to the following conditions:
301
+
302
+ The above copyright notice and this permission notice shall be included
303
+ in all copies or substantial portions of the Software.
304
+
305
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
306
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
307
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
308
+ THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
309
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
310
+ FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
311
+ DEALINGS IN THE SOFTWARE.
312
+
313
+ > plugin/multiscaleDeformableAttnPlugin/multiscaleDeformableAttn.cu
314
+ > plugin/multiscaleDeformableAttnPlugin/multiscaleDeformableAttn.h
315
+ > plugin/multiscaleDeformableAttnPlugin/multiscaleDeformableIm2ColCuda.cuh
316
+
317
+ Copyright 2020 SenseTime
318
+
319
+ Licensed under the Apache License, Version 2.0 (the "License");
320
+ you may not use this file except in compliance with the License.
321
+ You may obtain a copy of the License at
322
+
323
+ http://www.apache.org/licenses/LICENSE-2.0
324
+
325
+ Unless required by applicable law or agreed to in writing, software
326
+ distributed under the License is distributed on an "AS IS" BASIS,
327
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
328
+ See the License for the specific language governing permissions and
329
+ limitations under the License.
330
+
331
+ DETR
332
+
333
+ Copyright 2020 - present, Facebook, Inc
334
+
335
+ Licensed under the Apache License, Version 2.0 (the "License");
336
+ you may not use this file except in compliance with the License.
337
+ You may obtain a copy of the License at
338
+
339
+ http://www.apache.org/licenses/LICENSE-2.0
340
+
341
+ Unless required by applicable law or agreed to in writing, software
342
+ distributed under the License is distributed on an "AS IS" BASIS,
343
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
344
+ See the License for the specific language governing permissions and
345
+ limitations under the License.
346
+
347
+ > demo/Diffusion/utilities.py
348
+ > demo/Diffusion/stable_video_diffusion_pipeline.py
349
+
350
+ HuggingFace diffusers library.
351
+
352
+ Copyright 2024 The HuggingFace Team.
353
+
354
+ Licensed under the Apache License, Version 2.0 (the "License");
355
+ you may not use this file except in compliance with the License.
356
+ You may obtain a copy of the License at
357
+
358
+ http://www.apache.org/licenses/LICENSE-2.0
359
+
360
+ Unless required by applicable law or agreed to in writing, software
361
+ distributed under the License is distributed on an "AS IS" BASIS,
362
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
363
+ See the License for the specific language governing permissions and
364
+ limitations under the License.
365
+
366
+ > demo/Diffusion/utils_sd3/sd3_impls.py
367
+ > demo/Diffusion/utils_sd3/other_impls.py
368
+ > demo/Diffusion/utils_sd3/mmdit.py
369
+ > demo/Diffusion/stable_diffusion_3_pipeline.py
370
+
371
+ MIT License
372
+
373
+ Copyright (c) 2024 Stability AI
374
+
375
+ Permission is hereby granted, free of charge, to any person obtaining a copy
376
+ of this software and associated documentation files (the "Software"), to deal
377
+ in the Software without restriction, including without limitation the rights
378
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
379
+ copies of the Software, and to permit persons to whom the Software is
380
+ furnished to do so, subject to the following conditions:
381
+
382
+ The above copyright notice and this permission notice shall be included in all
383
+ copies or substantial portions of the Software.
384
+
385
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
386
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
387
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
388
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
389
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
390
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
391
+ SOFTWARE.
392
+
393
+ > demo/Diffusion/utilities.py
394
+
395
+ ModelScope library.
396
+
397
+ Copyright (c) Alibaba, Inc. and its affiliates.
398
+
399
+ Licensed under the Apache License, Version 2.0 (the "License");
400
+ you may not use this file except in compliance with the License.
401
+ You may obtain a copy of the License at
402
+
403
+ http://www.apache.org/licenses/LICENSE-2.0
404
+
405
+ Unless required by applicable law or agreed to in writing, software
406
+ distributed under the License is distributed on an "AS IS" BASIS,
407
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
408
+ See the License for the specific language governing permissions and
409
+ limitations under the License.
410
+
411
+ > plugin/scatterElementsPlugin/atomics.cuh
412
+ > plugin/scatterElementsPlugin/reducer.cuh
413
+ > plugin/scatterElementsPlugin/scatterElementsPluginKernel.cu
414
+ > plugin/scatterElementsPlugin/scatterElementsPluginKernel.h
415
+ > plugin/scatterElementsPlugin/TensorInfo.cuh
416
+
417
+ Copyright (c) 2020 Matthias Fey <matthias.fey@tu-dortmund.de>
418
+
419
+ Permission is hereby granted, free of charge, to any person obtaining a copy
420
+ of this software and associated documentation files (the "Software"), to deal
421
+ in the Software without restriction, including without limitation the rights
422
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
423
+ copies of the Software, and to permit persons to whom the Software is
424
+ furnished to do so, subject to the following conditions:
425
+
426
+ The above copyright notice and this permission notice shall be included in
427
+ all copies or substantial portions of the Software.
428
+
429
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
430
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
431
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
432
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
433
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
434
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
435
+ THE SOFTWARE.
436
+
437
+ > plugin/roiAlignPlugin/roiAlignKernel.cu
438
+
439
+ MIT License
440
+
441
+ Copyright (c) Microsoft Corporation
442
+
443
+ Permission is hereby granted, free of charge, to any person obtaining a copy
444
+ of this software and associated documentation files (the "Software"), to deal
445
+ in the Software without restriction, including without limitation the rights
446
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
447
+ copies of the Software, and to permit persons to whom the Software is
448
+ furnished to do so, subject to the following conditions:
449
+
450
+ The above copyright notice and this permission notice shall be included in all
451
+ copies or substantial portions of the Software.
452
+
453
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
454
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
455
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
456
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
457
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
458
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
459
+ SOFTWARE.
460
+
461
+ Project-URL: homepage, https://github.com/NVIDIA-NeMo/Eval
462
+ Project-URL: repository, https://github.com/NVIDIA-NeMo/Eval/packages/nemo-evaluator-launcher
463
+ Keywords: deep learning,evaluations,machine learning,gpu,NLP,pytorch,torch
464
+ Requires-Python: <3.14,>=3.10
465
+ Description-Content-Type: text/markdown
466
+ License-File: LICENSE
467
+ Requires-Dist: hydra-core<2.0.0,>=1.3.2
468
+ Requires-Dist: jinja2<4.0.0,>=3.1.6
469
+ Requires-Dist: leptonai>=0.25.0
470
+ Requires-Dist: pyyaml>=6.0.0
471
+ Requires-Dist: requests>=2.32.4
472
+ Requires-Dist: simple-parsing<0.2.0,>=0.1.7
473
+ Requires-Dist: structlog<26.0.0,>=25.4.0
474
+ Requires-Dist: tabulate<0.10.0,>=0.9.0
475
+ Requires-Dist: tomli<3.0.0,>=2.0.0; python_version < "3.11"
476
+ Provides-Extra: mlflow
477
+ Requires-Dist: mlflow>=2.8.0; extra == "mlflow"
478
+ Provides-Extra: wandb
479
+ Requires-Dist: wandb>=0.15.0; extra == "wandb"
480
+ Provides-Extra: gsheets
481
+ Requires-Dist: gsheets>=0.1.0; extra == "gsheets"
482
+ Provides-Extra: exporters
483
+ Requires-Dist: mlflow; extra == "exporters"
484
+ Requires-Dist: wandb; extra == "exporters"
485
+ Requires-Dist: gsheets; extra == "exporters"
486
+ Provides-Extra: all
487
+ Requires-Dist: mlflow; extra == "all"
488
+ Requires-Dist: wandb; extra == "all"
489
+ Requires-Dist: gsheets; extra == "all"
490
+ Dynamic: license-file
491
+
492
+ # NeMo Evaluator Launcher
493
+
494
+ For complete documentation, please see: [docs/nemo-evaluator-launcher/index.md](https://github.com/NVIDIA-NeMo/Eval/tree/main/docs/nemo-evaluator-launcher/index.md)
@@ -1,8 +1,5 @@
1
1
  [project]
2
2
  name = "nemo-evaluator-launcher"
3
- description = "NeMo Evaluator Launcher - Public API"
4
- requires-python = ">=3.10,<3.14"
5
- readme = "README.md"
6
3
  dependencies = [
7
4
  "hydra-core>=1.3.2,<2.0.0",
8
5
  "jinja2>=3.1.6,<4.0.0",
@@ -15,6 +12,30 @@ dependencies = [
15
12
  "tomli>=2.0.0,<3.0.0; python_version < '3.11'",
16
13
  ]
17
14
  dynamic = ["version"]
15
+ readme = "README.md"
16
+ description = "Launcher for the evaluations provided by NeMo Evaluator containers with different runtime backends"
17
+ requires-python = ">=3.10,<3.14"
18
+
19
+ # BEGIN(if-changed): check package_info.py, too
20
+ authors = [{ name = "NVIDIA" }, { email = "nemo-toolkit@nvidia.com" }]
21
+ maintainers = [{ name = "NVIDIA" }, { email = "nemo-toolkit@nvidia.com" }]
22
+ license = { file = "LICENSE" }
23
+ keywords = [
24
+ "deep learning",
25
+ "evaluations",
26
+ "machine learning",
27
+ "gpu",
28
+ "NLP",
29
+ "pytorch",
30
+ "torch",
31
+ ]
32
+ # END(if-changed)
33
+
34
+ [project.urls]
35
+ # BEGIN(if-changed): check package_info.py
36
+ homepage = "https://github.com/NVIDIA-NeMo/Eval"
37
+ repository = "https://github.com/NVIDIA-NeMo/Eval/packages/nemo-evaluator-launcher"
38
+ # END(if-changed)
18
39
 
19
40
  [project.optional-dependencies]
20
41
  mlflow = ["mlflow>=2.8.0"]
@@ -50,7 +71,11 @@ package-dir = { "" = "src" }
50
71
  where = ["src"]
51
72
 
52
73
  [tool.setuptools.package-data]
53
- "nemo_evaluator_launcher" = ["resources/**/*", "configs/**/*", "executors/**/*.sh"]
74
+ "nemo_evaluator_launcher" = [
75
+ "resources/**/*",
76
+ "configs/**/*",
77
+ "executors/**/*.sh",
78
+ ]
54
79
 
55
80
  [tool.setuptools.dynamic]
56
81
  version = { attr = "nemo_evaluator_launcher.package_info.__version__" }
@@ -18,8 +18,10 @@
18
18
  This module provides the main functional entry points for running evaluations, querying job status, and listing available tasks. These functions are intended to be used by CLI commands and external integrations.
19
19
  """
20
20
 
21
+ from pathlib import Path
21
22
  from typing import Any, List, Optional, Union
22
23
 
24
+ import yaml
23
25
  from omegaconf import DictConfig, OmegaConf
24
26
 
25
27
  from nemo_evaluator_launcher.api.types import RunConfig
@@ -434,7 +436,42 @@ def export_results(
434
436
  single_id = invocation_ids[0]
435
437
 
436
438
  if "." in single_id: # job_id
437
- job_data = ExecutionDB().get_job(single_id)
439
+ md_job_data = None
440
+ # Use artifacts/run_config.yml if present
441
+ ypath_artifacts = Path("run_config.yml")
442
+ if ypath_artifacts.exists():
443
+ try:
444
+ cfg_yaml = (
445
+ yaml.safe_load(ypath_artifacts.read_text(encoding="utf-8"))
446
+ or {}
447
+ )
448
+ # merge exporter config if present
449
+ ypath_export = Path("export_config.yml")
450
+ if ypath_export.exists():
451
+ exp_yaml = (
452
+ yaml.safe_load(ypath_export.read_text(encoding="utf-8"))
453
+ or {}
454
+ )
455
+ exec_cfg = cfg_yaml.get("execution") or {}
456
+ auto_exp = (exp_yaml.get("execution") or {}).get(
457
+ "auto_export"
458
+ )
459
+ if auto_exp is not None:
460
+ exec_cfg["auto_export"] = auto_exp
461
+ cfg_yaml["execution"] = exec_cfg
462
+ # metadata
463
+ md_job_data = JobData(
464
+ invocation_id=single_id.split(".")[0],
465
+ job_id=single_id,
466
+ timestamp=0.0,
467
+ executor="local", #
468
+ data={"output_dir": str(Path.cwd().parent)},
469
+ config=cfg_yaml,
470
+ )
471
+ except Exception:
472
+ md_job_data = None
473
+ # fallback to execDB only
474
+ job_data = md_job_data or ExecutionDB().get_job(single_id)
438
475
  if job_data is None:
439
476
  return {
440
477
  "success": False,
@@ -470,6 +470,8 @@ def _create_slurm_sbatch_script(
470
470
  s += "#SBATCH --exclusive\n"
471
471
  s += "#SBATCH --output {}\n".format(remote_task_subdir / "logs" / "slurm-%A.out")
472
472
  s += "\n"
473
+ s += f'TASK_DIR="{str(remote_task_subdir)}"\n'
474
+ s += "\n"
473
475
 
474
476
  # collect all env vars
475
477
  env_vars = copy.deepcopy(dict(cfg.evaluation.get("env_vars", {})))
@@ -620,11 +622,22 @@ def _generate_auto_export_section(
620
622
  if not destinations:
621
623
  return ""
622
624
 
623
- s = "\n# AUTO-EXPORT ON SUCCESS\n"
625
+ s = "\n# Auto-export on success\n"
624
626
  s += "EVAL_EXIT_CODE=$?\n"
625
627
  s += "if [ $EVAL_EXIT_CODE -eq 0 ]; then\n"
626
628
  s += " echo 'Evaluation completed successfully. Starting auto-export...'\n"
627
-
629
+ s += " set +e\n" # per exporter failure allowed
630
+ s += " set +x\n"
631
+ s += ' cd "$TASK_DIR/artifacts"\n'
632
+ auto_export_cfg = OmegaConf.to_container(
633
+ cfg.execution.get("auto_export", {}), resolve=True
634
+ )
635
+ yaml_str = yaml.safe_dump(
636
+ {"execution": {"auto_export": auto_export_cfg}}, sort_keys=False
637
+ )
638
+ s += " cat > export_config.yml << 'EOF'\n"
639
+ s += yaml_str
640
+ s += "EOF\n"
628
641
  for dest in destinations:
629
642
  s += f" echo 'Exporting to {dest}...'\n"
630
643
  s += f" nemo-evaluator-launcher export {job_id} --dest {dest} || echo 'Export to {dest} failed'\n"
@@ -13,10 +13,11 @@
13
13
  # limitations under the License.
14
14
 
15
15
 
16
+ # Below is the _next_ version that will be published, not the currently published one.
16
17
  MAJOR = 0
17
18
  MINOR = 1
18
19
  PATCH = 0
19
- PRE_RELEASE = "rc7"
20
+ PRE_RELEASE = "rc9"
20
21
 
21
22
  # Use the following formatting: (major, minor, patch, pre-release)
22
23
  VERSION = (MAJOR, MINOR, PATCH, PRE_RELEASE)
@@ -24,6 +25,7 @@ VERSION = (MAJOR, MINOR, PATCH, PRE_RELEASE)
24
25
  __shortversion__ = ".".join(map(str, VERSION[:3]))
25
26
  __version__ = ".".join(map(str, VERSION[:3])) + "".join(VERSION[3:])
26
27
 
28
+ # BEGIN(if-changed): check the pyproject.toml, too
27
29
  __package_name__ = "nemo_evaluator_launcher"
28
30
  __contact_names__ = "NVIDIA"
29
31
  __contact_emails__ = "nemo-toolkit@nvidia.com"
@@ -33,3 +35,4 @@ __download_url__ = "https://github.com/NVIDIA-NeMo/Eval/releases"
33
35
  __description__ = "Launcher for the evaluations provided by NeMo Evaluator containers with different runtime backends"
34
36
  __license__ = "Apache2"
35
37
  __keywords__ = "deep learning, evaluations, machine learning, gpu, NLP, pytorch, torch"
38
+ # END(if-changed)