|
408 | 408 | "\n", |
409 | 409 | "if RELAXATION_KGRID is not None and ADD_RELAXATION:\n", |
410 | 410 | " unit = workflow.subworkflows[0].get_unit_by_name(name_regex=\"relax\")\n", |
411 | | - " unit.add_context(PointsGridDataProvider(dimensions=RELAXATION_KGRID, isEdited=True).yield_data())\n", |
| 411 | + " unit.add_context(PointsGridDataProvider(dimensions=RELAXATION_KGRID, isEdited=True).get_context_item_data())\n", |
412 | 412 | " workflow.subworkflows[0].set_unit(unit)\n", |
413 | 413 | "\n", |
414 | 414 | "if SCF_KGRID is not None:\n", |
415 | 415 | " unit = preliminary_scf_subworkflow.get_unit_by_name(name=\"pw_scf\")\n", |
416 | 416 | " if unit:\n", |
417 | | - " unit.add_context(PointsGridDataProvider(dimensions=SCF_KGRID, isEdited=True).yield_data())\n", |
| 417 | + " unit.add_context(PointsGridDataProvider(dimensions=SCF_KGRID, isEdited=True).get_context_item_data())\n", |
418 | 418 | " preliminary_scf_subworkflow.set_unit(unit)\n", |
419 | 419 | "\n", |
420 | 420 | " unit = main_hse_subworkflow.get_unit_by_name(name=\"pw_scf_bands_hse\")\n", |
421 | 421 | " if unit:\n", |
422 | | - " unit.add_context(PointsGridDataProvider(dimensions=SCF_KGRID, isEdited=True).yield_data())\n", |
423 | | - " unit.add_context(PointsGridDataProvider(name=\"qgrid\", dimensions=SCF_KGRID, isEdited=True).yield_data())\n", |
| 422 | + " unit.add_context(PointsGridDataProvider(dimensions=SCF_KGRID, isEdited=True).get_context_item_data())\n", |
| 423 | + " unit.add_context(PointsGridDataProvider(name=\"qgrid\", dimensions=SCF_KGRID, isEdited=True).get_context_item_data())\n", |
424 | 424 | " main_hse_subworkflow.set_unit(unit)\n", |
425 | 425 | "\n", |
426 | 426 | "if KPATH is not None:\n", |
427 | 427 | " unit = main_hse_subworkflow.get_unit_by_name(name=\"pw_scf_bands_hse\")\n", |
428 | 428 | " if unit:\n", |
429 | | - " unit.add_context(PointsPathDataProvider(path=KPATH, isEdited=True).yield_data())\n", |
| 429 | + " unit.add_context(PointsPathDataProvider(path=KPATH, isEdited=True).get_context_item_data())\n", |
430 | 430 | " main_hse_subworkflow.set_unit(unit)\n", |
431 | 431 | "\n", |
432 | 432 | "if ECUTWFC is not None:\n", |
433 | | - " cutoffs_context = PlanewaveCutoffsContextProvider(wavefunction=ECUTWFC, density=ECUTRHO, isEdited=True).yield_data()\n", |
| 433 | + " cutoffs_context = PlanewaveCutoffsContextProvider(wavefunction=ECUTWFC, density=ECUTRHO, isEdited=True).get_context_item_data()\n", |
434 | 434 | " for unit_name in [\"pw_relax\", \"pw_vc-relax\", \"pw_scf_bands_hse\"]:\n", |
435 | 435 | " for swf in workflow.subworkflows:\n", |
436 | 436 | " unit = swf.get_unit_by_name(name=unit_name)\n", |
|
0 commit comments