Skip to content

Commit 3b6beb4

Browse files
DarinochkaDarinochka
committed
fix: test
1 parent 8a5d015 commit 3b6beb4

File tree

1 file changed

+148
-5
lines changed

1 file changed

+148
-5
lines changed

‎tests/callback/test_callback.py‎

Lines changed: 148 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,29 @@ def test_pipeline_callbacks(dataset):
103103
"num": 0,
104104
},
105105
),
106-
("log_metric", {"metrics": {"retrieval_hit_rate": 1.0}}),
106+
(
107+
"log_metric",
108+
{
109+
"metrics": {
110+
"retrieval_hit_rate": 1.0,
111+
"emissions/cpu_count": 4,
112+
"emissions/cpu_energy": 0.00010161774583333391,
113+
"emissions/cpu_power": 140.0,
114+
"emissions/duration": -1743448871.1596549,
115+
"emissions/emissions": 5.0399069483124444e-05,
116+
"emissions/emissions_rate": 1.9287459668038395e-05,
117+
"emissions/energy_consumed": 0.00010586762294121488,
118+
"emissions/gpu_energy": 0,
119+
"emissions/gpu_power": 0.0,
120+
"emissions/latitude": 29.4227,
121+
"emissions/longitude": -98.4927,
122+
"emissions/pue": 1.0,
123+
"emissions/ram_energy": 4.249877107880973e-06,
124+
"emissions/ram_power": 5.8557257652282715,
125+
"emissions/ram_total_size": 15.61526870727539,
126+
}
127+
},
128+
),
107129
("end_module", {}),
108130
(
109131
"start_module",
@@ -113,7 +135,29 @@ def test_pipeline_callbacks(dataset):
113135
"num": 1,
114136
},
115137
),
116-
("log_metric", {"metrics": {"retrieval_hit_rate": 1.0}}),
138+
(
139+
"log_metric",
140+
{
141+
"metrics": {
142+
"retrieval_hit_rate": 1.0,
143+
"emissions/cpu_count": 4,
144+
"emissions/cpu_energy": 4.869346412777829e-05,
145+
"emissions/cpu_power": 140.0,
146+
"emissions/duration": -1743448872.5204384,
147+
"emissions/emissions": 2.415026452774112e-05,
148+
"emissions/emissions_rate": 1.9287193027812115e-05,
149+
"emissions/energy_consumed": 5.072972825043102e-05,
150+
"emissions/gpu_energy": 0,
151+
"emissions/gpu_power": 0.0,
152+
"emissions/latitude": 29.4227,
153+
"emissions/longitude": -98.4927,
154+
"emissions/pue": 1.0,
155+
"emissions/ram_energy": 2.036264122652723e-06,
156+
"emissions/ram_power": 5.8557257652282715,
157+
"emissions/ram_total_size": 15.61526870727539,
158+
}
159+
},
160+
),
117161
("end_module", {}),
118162
(
119163
"start_module",
@@ -139,7 +183,30 @@ def test_pipeline_callbacks(dataset):
139183
"num": 0,
140184
},
141185
),
142-
("log_metric", {"metrics": {"scoring_accuracy": 1.0, "scoring_roc_auc": 1.0}}),
186+
(
187+
"log_metric",
188+
{
189+
"metrics": {
190+
"emissions/cpu_count": 4,
191+
"emissions/cpu_energy": 5.6463417383332314e-05,
192+
"emissions/cpu_power": 140.0,
193+
"emissions/duration": -1743448872.3206754,
194+
"emissions/emissions": 2.800390409611645e-05,
195+
"emissions/emissions_rate": 1.9287278349336493e-05,
196+
"emissions/energy_consumed": 5.882463287784108e-05,
197+
"emissions/gpu_energy": 0,
198+
"emissions/gpu_power": 0.0,
199+
"emissions/latitude": 29.4227,
200+
"emissions/longitude": -98.4927,
201+
"emissions/pue": 1.0,
202+
"emissions/ram_energy": 2.3612154945087627e-06,
203+
"emissions/ram_power": 5.8557257652282715,
204+
"emissions/ram_total_size": 15.61526870727539,
205+
"scoring_accuracy": 1.0,
206+
"scoring_roc_auc": 1.0,
207+
}
208+
},
209+
),
143210
("end_module", {}),
144211
(
145212
"start_module",
@@ -165,7 +232,30 @@ def test_pipeline_callbacks(dataset):
165232
"num": 1,
166233
},
167234
),
168-
("log_metric", {"metrics": {"scoring_accuracy": 1.0, "scoring_roc_auc": 1.0}}),
235+
(
236+
"log_metric",
237+
{
238+
"metrics": {
239+
"emissions/cpu_count": 4,
240+
"emissions/cpu_energy": 5.822425206111052e-05,
241+
"emissions/cpu_power": 140.0,
242+
"emissions/duration": -1743448872.2755105,
243+
"emissions/emissions": 2.8877239803966572e-05,
244+
"emissions/emissions_rate": 1.9287335939110744e-05,
245+
"emissions/energy_consumed": 6.065915038715212e-05,
246+
"emissions/gpu_energy": 0,
247+
"emissions/gpu_power": 0.0,
248+
"emissions/latitude": 29.4227,
249+
"emissions/longitude": -98.4927,
250+
"emissions/pue": 1.0,
251+
"emissions/ram_energy": 2.434898326041607e-06,
252+
"emissions/ram_power": 5.8557257652282715,
253+
"emissions/ram_total_size": 15.61526870727539,
254+
"scoring_accuracy": 1.0,
255+
"scoring_roc_auc": 1.0,
256+
}
257+
},
258+
),
169259
("end_module", {}),
170260
(
171261
"start_module",
@@ -189,7 +279,30 @@ def test_pipeline_callbacks(dataset):
189279
"num": 0,
190280
},
191281
),
192-
("log_metric", {"metrics": {"scoring_accuracy": 0.75, "scoring_roc_auc": 1.0}}),
282+
(
283+
"log_metric",
284+
{
285+
"metrics": {
286+
"emissions/cpu_count": 4,
287+
"emissions/cpu_energy": 7.098722081111014e-05,
288+
"emissions/cpu_power": 140.0,
289+
"emissions/duration": -1743448871.9471838,
290+
"emissions/emissions": 3.520725453309416e-05,
291+
"emissions/emissions_rate": 1.928735273284849e-05,
292+
"emissions/energy_consumed": 7.39558961292537e-05,
293+
"emissions/gpu_energy": 0,
294+
"emissions/gpu_power": 0.0,
295+
"emissions/latitude": 29.4227,
296+
"emissions/longitude": -98.4927,
297+
"emissions/pue": 1.0,
298+
"emissions/ram_energy": 2.968675318143547e-06,
299+
"emissions/ram_power": 5.8557257652282715,
300+
"emissions/ram_total_size": 15.61526870727539,
301+
"scoring_accuracy": 0.75,
302+
"scoring_roc_auc": 1.0,
303+
}
304+
},
305+
),
193306
("end_module", {}),
194307
("start_module", {"module_kwargs": {"thresh": 0.5}, "module_name": "threshold", "num": 0}),
195308
(
@@ -201,6 +314,21 @@ def test_pipeline_callbacks(dataset):
201314
"decision_precision": 0.55,
202315
"decision_recall": 0.8,
203316
"decision_roc_auc": 0.8428571428571429,
317+
"emissions/cpu_count": 4,
318+
"emissions/cpu_energy": 4.606736166661247e-07,
319+
"emissions/cpu_power": 140.0,
320+
"emissions/duration": -1743448873.7607834,
321+
"emissions/emissions": 2.2824627343844443e-07,
322+
"emissions/emissions_rate": 1.9233611973570262e-05,
323+
"emissions/energy_consumed": 4.794511220531496e-07,
324+
"emissions/gpu_energy": 0,
325+
"emissions/gpu_power": 0.0,
326+
"emissions/latitude": 29.4227,
327+
"emissions/longitude": -98.4927,
328+
"emissions/pue": 1.0,
329+
"emissions/ram_energy": 1.8777505387024907e-08,
330+
"emissions/ram_power": 5.8557257652282715,
331+
"emissions/ram_total_size": 15.61526870727539,
204332
}
205333
},
206334
),
@@ -215,6 +343,21 @@ def test_pipeline_callbacks(dataset):
215343
"decision_precision": 0.55,
216344
"decision_recall": 0.8,
217345
"decision_roc_auc": 0.8428571428571429,
346+
"emissions/cpu_count": 4,
347+
"emissions/cpu_energy": 4.5328064444384436e-07,
348+
"emissions/cpu_power": 140.0,
349+
"emissions/duration": -1743448873.7610447,
350+
"emissions/emissions": 2.2460812015265647e-07,
351+
"emissions/emissions_rate": 1.9237634797985535e-05,
352+
"emissions/energy_consumed": 4.7180886507872054e-07,
353+
"emissions/gpu_energy": 0,
354+
"emissions/gpu_power": 0.0,
355+
"emissions/latitude": 29.4227,
356+
"emissions/longitude": -98.4927,
357+
"emissions/pue": 1.0,
358+
"emissions/ram_energy": 1.852822063487629e-08,
359+
"emissions/ram_power": 5.8557257652282715,
360+
"emissions/ram_total_size": 15.61526870727539,
218361
}
219362
},
220363
),

0 commit comments

Comments
 (0)