Ready for Dataiku 10? Try out the Crash Course on new features!GET STARTED

save pickle in managed folder

Isma
Level 1
save pickle in managed folder

Hello,

I want to save a picke object in DSS folder. Here is an example of working code outside DSS. How can I use the DSS API to do so in DSS.

def save_model(self😞
# Names
scope=self.scope.replace(' ', '_').lower()
model_name='model_' + scope
opt_tree_name='opt_tree_' + scope
pkl_name=model_name + '.pkl'
# Save pickles
with open(self.builds_path + pkl_name, 'wb') as f:
pickle.dump(self, f)
0 Kudos
1 Reply
AlexandreL
Dataiker
Dataiker

Hi,

You can use the folder api to get a writer:

with dataiku.Folder(FOLDER_ID).get_writer(MODEL_PATH_IN_FOLDER) as writer:
	writeable = pickle.dumps(MODEL)
	writer.write(writeable)

 

A banner prompting to get Dataiku DSS