Skip to content

Conversation

@Sukh-P
Copy link
Member

@Sukh-P Sukh-P commented Nov 12, 2025

Pull Request

Description

Changes to support one torch dataset/input data schema this is a development branch which has had two major commits into it which were already reviewed, those were:

Fixes #334

How Has This Been Tested?

Through unit tests and also running training with the new changes

@Sukh-P Sukh-P changed the title Dev one torch dataset one torch dataset & target data schema Nov 13, 2025
@Sukh-P Sukh-P marked this pull request as ready for review November 13, 2025 16:38
location_id: id for location
"""
# Check the user has asked for a sample which we have the data for
if t0 not in self.valid_t0_times or self.valid_t0_and_location_ids:
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think this might need a little more care. If self.complete_generation==True we can check for t0 in self.complete_generation but I think for self.complete_generation==False we should really check if the pair (t0, location_id) is in self.valid_t0_and_location_ids.

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yep that's fair, I've added a stronger check now

@Sukh-P
Copy link
Member Author

Sukh-P commented Nov 14, 2025

Thanks for taking a look at this @dfulu, I have addressed all the comments now so hopefully this is good to go

Copy link
Member

@dfulu dfulu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks great! Thanks Sukh!

@Sukh-P Sukh-P merged commit 70edc01 into main Nov 21, 2025
6 checks passed
@Sukh-P Sukh-P deleted the dev-one-torch-dataset branch November 21, 2025 10:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Join GSP and Site pipelines together

3 participants